update

app.py

@@ -51,7 +51,7 @@ def accelerate_train_lora(steps):
       --learning_rate=1e-4 \
       --lr_scheduler="constant" \
       --lr_warmup_steps=0 \
-      --max_train_steps={steps}') #,shell=True) #30000
+      --max_train_steps={int(steps)}') #,shell=True) #30000
     print("***********inside accelerate_train_lora 22222***********")
     return
     
@@ -61,11 +61,13 @@ with gr.Blocks() as demo:
         #in_prompt = gr.Textbox(label="Enter a ")
         in_steps = gr.Number(label="Enter number of steps")
         in_alpha = gr.Slider(0.1,1.0, step=0.01, label="Set Alpha level - higher value has more chances to overfit")
+    with gr.Row():
         b1 = gr.Button(value="Train LORA model")
         b2 = gr.Button(value="Inference using LORA model")
     with gr.Row():
         out_image = gr.Image(label="Image generated by LORA model")
-    b1.click(fn = accelerate_train_lora, inputs=in_steps)
+        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, outputs=out_image)
 
 demo.launch(debug=True, show_error=True)