Update app.py

app.py

@@ -37,10 +37,16 @@ with gr.Blocks() as demo:
     dataset_predefined = gr.Dropdown(label="Predefined Dataset", choices=['1', '2', '3'], value='1', visible=True)
     dataset_upload = gr.File(label="Upload Dataset", visible=False)
     
-    # Update visibility of dataset options based on user choice
+    # Function to update visibility based on user choice
     def update_dataset_visibility(choice):
-        dataset_predefined.visible = (choice == "Predefined Dataset")
-        dataset_upload.visible = (choice == "Upload Your Own")
+        if choice == "Predefined Dataset":
+            dataset_predefined.visible = True
+            dataset_upload.visible = False
+            return dataset_predefined, dataset_upload
+        elif choice == "Upload Your Own":
+            dataset_predefined.visible = False
+            dataset_upload.visible = True
+            return dataset_predefined, dataset_upload
     
     dataset_choice.change(update_dataset_visibility, inputs=[dataset_choice], outputs=[dataset_predefined, dataset_upload])