Update app.py

app.py

@@ -33,25 +33,21 @@ with gr.Blocks() as demo:
     # Prompt Input
     name_input = gr.Textbox(label="Your Name")
     # Dataset choice
-    dataset_choice = gr.Radio(label="Choose Dataset", choices=["Predefined Dataset", "Upload Your Own"], value="Upload Your Own")
-    dataset_predefined = gr.Dropdown(label="Predefined Dataset", choices=['1', '2', '3'], value='1', visible=False)
-    dataset_upload = gr.File(label="Upload Dataset", visible=True)
+    dataset_choice = gr.Radio(label="Choose Dataset", choices=["Predefined Dataset", "Upload Your Own"], value="Predefined Dataset")
+    dataset_predefined = gr.Dropdown(label="Predefined Dataset", choices=['1', '2', '3'], value='1', visible=True)
+    dataset_upload = gr.Upload(label="Upload Dataset", accept=".csv,.txt", visible=False)
     
-    # Function to update visibility based on user choice
-    def update_dataset_visibility(choice):
+    # Function to toggle visibility based on user choice
+    def toggle_dataset_visibility(choice):
         if choice == "Predefined Dataset":
             dataset_predefined.visible = True
             dataset_upload.visible = False
         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)
-    
-    # Update visibility based on user choice
-    dataset_choice.change(update_dataset_visibility, inputs=[dataset_choice], outputs=[dataset_predefined, dataset_upload])
+    # Initial visibility setup based on default choice
+    toggle_dataset_visibility(dataset_choice.value)
     
     ##### Model Outputs #####
 
@@ -64,9 +60,12 @@ with gr.Blocks() as demo:
     tune_btn = gr.Button("Start Fine Tuning")
     # Execute button
     tune_btn.click(fn=greet, 
-                   inputs=[model_name, prompt_template, name_input, dataset_predefined],
+                   inputs=[model_name, prompt_template, name_input, dataset_upload],
                    outputs=output)
 
+    # Update visibility based on user choice
+    dataset_choice.change(toggle_dataset_visibility, inputs=[dataset_choice], outputs=[])
+
 ##################################### Launch #######################################
 
 if __name__ == "__main__":