upd

app.py

@@ -53,7 +53,7 @@ def accelerate_train_lora(steps):
       --lr_warmup_steps=0 \
       --max_train_steps={int(steps)}') #,shell=True) #30000
     print("*********** completing accelerate_train_lora ***********")
-    return
+    return "./output_example/lora_weight.pt"
     
 with gr.Blocks() as demo:
     with gr.Row():
@@ -67,7 +67,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         in_prompt = gr.Textbox(label="Enter a prompt for fine-tuned LORA model", visible=True)
         out_image = gr.Image(label="Image generated by LORA model")
-        out_file = gr.File(label="Lora trained model weights")
+        out_file = gr.File(label="Lora trained model weights", )
     b1.click(fn = accelerate_train_lora, inputs=in_steps, outputs=out_file)
     b2.click(fn = monkeypatching, inputs=[in_alpha, in_prompt], outputs=out_image)