update

app.py

@@ -68,20 +68,15 @@ with gr.Blocks() as demo:
             b1 = gr.Button(value="Train LORA model")
             in_prompt = gr.Textbox(label="Enter a prompt for fine-tuned LORA model", visible=True)
             b2 = gr.Button(value="Inference using LORA model")
-            #with gr.Accordion("Advance settings", open=False):
-            #    gr.Markdown("Advance settings for a number of Training Steps and Alpha. Set alpha to 1.0 to fully add LORA. If the LORA seems to have too much effect (i.e., overfitting), set alpha to a lower value. If the LORA seems to have too little effect, set the alpha higher. You can tune these two values to your needs.")
-            #    in_steps = gr.Number(label="Enter the number of training steps", value = 4000)
-            #    in_alpha = gr.Slider(0.1,1.0, step=0.01, label="Set Alpha level", value=0.5)
+
     with gr.Row():
-        #with gr.Column():
-            #in_prompt = gr.Textbox(label="Enter a prompt for fine-tuned LORA model", visible=True)
-            #b2 = gr.Button(value="Inference using LORA model")
         out_image = gr.Image(label="Image generated by LORA model")
-        out_file = gr.File(label="Lora trained model weights", )
-        with gr.Accordion("Advance settings", open=False):
-            gr.Markdown("Advance settings for a number of Training Steps and Alpha. Set alpha to 1.0 to fully add LORA. If the LORA seems to have too much effect (i.e., overfitting), set alpha to a lower value. If the LORA seems to have too little effect, set the alpha higher. You can tune these two values to your needs.")
-            in_steps = gr.Number(label="Enter the number of training steps", value = 4000)
-            in_alpha = gr.Slider(0.1,1.0, step=0.01, label="Set Alpha level", value=0.5)
+        with gr.Column():
+            out_file = gr.File(label="Lora trained model weights", )
+            with gr.Accordion("Advance settings for Training and Inference", open=False):
+                gr.Markdown("Advance settings for a number of Training Steps and Alpha. Set alpha to 1.0 to fully add LORA. If the LORA seems to have too much effect (i.e., overfitting), set alpha to a lower value. If the LORA seems to have too little effect, set the alpha higher. You can tune these two values to your needs.")
+                in_steps = gr.Number(label="Enter the number of training steps", value = 4000)
+                in_alpha = gr.Slider(0.1,1.0, step=0.01, label="Set Alpha level", value=0.5)
         
     b1.click(fn = accelerate_train_lora, inputs=in_steps, outputs=out_file)
     b2.click(fn = monkeypatching, inputs=[in_alpha, in_prompt], outputs=out_image)