Update app.py

app.py

@@ -24,7 +24,7 @@ def upload_visibility(radio):
     else:
         return gr.UploadButton(visible=bool(0))
 
-def greet(model_name, prompt_template, name, dataset):
+def greet(model_name, inject_prompt, name, dataset):
     """The model call"""
     return f"Hello {name}!! Using model: {model_name} with template: {inject_prompt}"
 
@@ -64,7 +64,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_predefined],
+                   inputs=[model_name, inject_prompt, name_input, dataset_predefined],
                    outputs=output)
 
 ##################################### Launch #######################################