Update app.py

app.py

@@ -45,9 +45,10 @@ with gr.Blocks() as demo:
         elif choice == "Upload Your Own":
             dataset_predefined.visible = False
             dataset_upload.visible = True
+        return dataset_predefined, dataset_upload  # Return both components
     
     # Initial call to set visibility based on default choice
-    update_dataset_visibility(dataset_choice.value)
+    dataset_predefined, dataset_upload = update_dataset_visibility(dataset_choice.value)
     
     # Update visibility based on user choice
     dataset_choice.change(update_dataset_visibility, inputs=[dataset_choice], outputs=[dataset_predefined, dataset_upload])