updated interface

app.py

@@ -13,17 +13,17 @@ from sklearn.model_selection import cross_val_score
 
 def compute(file, y_attribute, cv_folds):
     if file is None:
-        return 'Please upload a .arff file'
+        return 'Please upload a .arff file', y_attribute
     if file.name.endswith('.arff'):
         dataset = openml.datasets.OpenMLDataset('t', 'test', data_file=file.name)
         X_, _, categorical_indicator_, attribute_names_ = dataset.get_data(
             dataset_format="array")
         if y_attribute not in attribute_names_:
-            return f"**Select attribute from {', '.join(attribute_names_)}**"
+            return f"**Select attribute from {', '.join(attribute_names_)}**", y_attribute
         X, y, categorical_indicator_, attribute_names_ = dataset.get_data(
             dataset_format="array", target=y_attribute)
     else:
-        return 'Please upload a .arff file'
+        return 'Please upload a .arff file', y_attribute
 
     order = np.arange(y.shape[0])
     np.random.seed(13)
@@ -40,7 +40,7 @@ def compute(file, y_attribute, cv_folds):
     # print(file, type(file))
     return f"ROC AUC OVO Cross Val mean is {sum(scores) / len(scores)} from {scores}. " + (
         "The PFN is only trained for datasets with up to 1024 training examples and it had to extrapolate to greater datasets for this evaluation." if len(
-            y) // cv_folds > 1024 else "")
+            y) // cv_folds > 1024 else ""), y_attribute
 
 
 def upload_file(file):
@@ -58,7 +58,7 @@ def upload_file(file):
 
 with gr.Blocks() as demo:
     gr.Markdown("""This demo allows you to play with the **TabPFN**.
-    Upload a .arff file, select an attribute to predict and the number of cross validation folds and get the ROC AUC OVO score.
+    Upload a .arff file, select an attribute to predict and the number of cross validation folds and get the ROC AUC OVO score for one seed.
     """)
     inp_file = gr.File(
         label='Drop a .arff file.')
@@ -67,7 +67,7 @@ with gr.Blocks() as demo:
 
     y_attribute = gr.Textbox(label='y attribute')
 
-    examples = gr.Examples(examples=['iris.csv', 'balance-scale.arff'],
+    examples = gr.Examples(examples=['balance-scale.arff'],
                            inputs=[inp_file],
                            outputs=[out_text, y_attribute],
                            fn=upload_file,
@@ -76,6 +76,6 @@ with gr.Blocks() as demo:
     # out_table = gr.DataFrame()
     inp_file.change(fn=upload_file, inputs=inp_file, outputs=[out_text, y_attribute])
 
-    btn.click(fn=compute, inputs=[inp_file, y_attribute, cv_folds], outputs=[out_text])
+    btn.click(fn=compute, inputs=[inp_file, y_attribute, cv_folds], outputs=[out_text, y_attribute])
 
 demo.launch()