Update app.py

app.py

@@ -169,12 +169,8 @@ def clear_directory(directory_path):
     else:
         print(f'The directory {directory_path} does not exist.')
 
-def get_start_info(b_lora_name):
-    your_username = api.whoami(token=hf_token)["name"]
-    return gr.update(visible=True, value=f"https://hf.co/{your_username}/{b_lora_name}"), gr.update(visible=True)
-
-def main(image_path, b_lora_trained_folder, instance_prompt, training_type, training_steps):
-
+def get_start_info(image_path, b_lora_name, instance_prompt):
+    
     if is_shared_ui:
         raise gr.Error("This Space only works in duplicated instances")
 
@@ -184,13 +180,18 @@ def main(image_path, b_lora_trained_folder, instance_prompt, training_type, trai
     if image_path == None:
         raise gr.Error("You forgot to specify an image reference")
 
-    if b_lora_trained_folder == "":
+    if b_lora_name == "":
         raise gr.Error("You forgot to specify a name for you model")
 
     if instance_prompt == "":
         raise gr.Error("You forgot to specify an instance prompt")
-
     
+    your_username = api.whoami(token=hf_token)["name"]
+    
+    return gr.update(visible=True, value=f"https://hf.co/{your_username}/{b_lora_name}"), gr.update(visible=True)
+
+def main(image_path, b_lora_trained_folder, instance_prompt, training_type, training_steps):
+
     local_dir = "image_to_train"
     # Check if the directory exists and create it if necessary
     if not os.path.exists(local_dir):
@@ -381,7 +382,7 @@ with gr.Blocks(css=css) as demo:
     
     train_btn.click(
         fn = get_start_info,
-        inputs = [b_lora_name],
+        inputs = [image, b_lora_name, instance_prompt],
         outputs = [started_info, status]
     ).then(
         fn = main,