Update app.py

app.py

@@ -15,7 +15,8 @@ def get_json_cfg():
 
 conf = get_json_cfg()
 
-def greet(model_name, prompt_template, name, dataset):
+def greet(model_name, prompt_template, name, dataset_file):
+    # Here you can process the uploaded file (dataset_file) as needed
     return f"Hello {name}!! Using model: {model_name} with template: {prompt_template}"
 
 ##################################### App UI #######################################
@@ -35,16 +36,16 @@ with gr.Blocks() as demo:
     # Dataset choice
     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)
+    dataset_file = gr.inputs.File(label="Upload Dataset", type='file', accept=".csv,.txt", visible=False)
     
     # 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
+            dataset_file.visible = False
         elif choice == "Upload Your Own":
             dataset_predefined.visible = False
-            dataset_upload.visible = True
+            dataset_file.visible = True
     
     # Initial visibility setup based on default choice
     toggle_dataset_visibility(dataset_choice.value)
@@ -60,7 +61,7 @@ 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_upload],
+                   inputs=[model_name, prompt_template, name_input, dataset_file],
                    outputs=output)
 
     # Update visibility based on user choice