Update app.py

app.py

@@ -25,6 +25,7 @@ def run_lora(prompt):
     image_bytes = query(payload, api_url, token)
     return Image.open(image_bytes)
 
+
 # Gradio UI
 print("Before Gradio Interface")
 
@@ -40,10 +41,10 @@ with gr.Blocks(css="custom.css") as app:
     result = gr.Image(interactive=False, label="Generated Image")
 
     with gr.Row():
-        with gr.Column(width=6):  # Adjusting the width
+        with gr.Column():  # Removed the width parameter
             title
             gallery
-        with gr.Column(width=6):  # Adjusting the width
+        with gr.Column():  # Removed the width parameter
             prompt
             gr.Button("Run").click(
                 fn=run_lora,
@@ -54,5 +55,5 @@ with gr.Blocks(css="custom.css") as app:
 
 print("After Gradio Interface")
 
-# Launch the Gradio interface with a queue
-app.launch(debug=True)
+# Launch the Gradio interface with a queue and specify the width here
+app.launch(debug=True, layout=("100%", "100%"))  # Added layout parameter