Update app.py

app.py

@@ -10,9 +10,9 @@ from utilities.templates import prompt_template
 
 conf = get_json_cfg()
 
-def dropdown_visibility(radio):
+def textbox_visibility(radio):
     value = radio
-    if value == "Predefined Dataset":
+    if value == "Hugging Face Hub Dataset":
         return gr.Dropdown(visible=bool(1))
     else:
         return gr.Dropdown(visible=bool(0))
@@ -24,9 +24,9 @@ def upload_visibility(radio):
     else:
         return gr.UploadButton(visible=bool(0))
 
-def greet(model_name, inject_prompt, name, dataset):
+def greet(model_name, inject_prompt, dataset):
     """The model call"""
-    return f"Hello {name}!! Using model: {model_name} with template: {inject_prompt}"
+    return f"Hello!! Using model: {model_name} with template: {inject_prompt}"
 
 ##################################### App UI #######################################
 with gr.Blocks() as demo:
@@ -44,12 +44,10 @@ with gr.Blocks() as demo:
     # Prompt template
     inject_prompt = gr.Textbox(label="Prompt Template", 
                                  value=prompt_template())
-    # 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="Predefined Dataset")
-    dataset_predefined = gr.Dropdown(label="Predefined Dataset", choices=['1', '2', '3'], value='1', visible=True)
-    dataset_upload = gr.UploadButton(label="Upload Dataset", file_types=[".pdf",".csv",".jsonl"], visible=False) # gr.File(label="Upload Dataset", visible=False)
+    dataset_choice = gr.Radio(label="Choose Dataset", choices=["Hugging Face Hub Dataset", "Upload Your Own"], value="Hugging Face Hub Dataset")
+    dataset_predefined = gr.Textbox(label="Hugging Face Hub Dataset", value='yahma/alpaca-cleaned', visible=True)
+    dataset_upload = gr.UploadButton(label="Upload Dataset (csv, jsonl, or txt)", file_types=[".csv",".jsonl", ".txt"], visible=False)
     dataset_choice.change(dropdown_visibility, dataset_choice, dataset_predefined)
     dataset_choice.change(upload_visibility, dataset_choice, dataset_upload)
 
@@ -64,7 +62,7 @@ with gr.Blocks() as demo:
     tune_btn = gr.Button("Start Fine Tuning")
     # Execute button
     tune_btn.click(fn=greet, 
-                   inputs=[model_name, inject_prompt, name_input, dataset_predefined],
+                   inputs=[model_name, inject_prompt, dataset_predefined],
                    outputs=output)
 
 ##################################### Launch #######################################