Update app.py

app.py

@@ -69,7 +69,7 @@ def replace_hf_profile(base_model_id):
 def process(action, base_model_id, dataset, system_prompt, user_prompt, schema):
     #raise gr.Error("Please clone and bring your own credentials.")
     if action == action_1:
-        result = fine_tune_model(base_model_id)
+        result = fine_tune_model(base_model_id, dataset)
     elif action == action_2:
         fine_tuned_model_id = replace_hf_profile(base_model_id)
         result = prompt_model(fine_tuned_model_id, system_prompt, user_prompt, schema)