Update app.py

app.py

@@ -17,36 +17,40 @@ def query(payload, api_url, token):
 
 # Define the function to run when the button is clicked
 def run_lora(prompt):
-    selected_lora = loras[0]  # You can change this to select a different LoRA if needed
+    selected_lora = loras[0]
     api_url = f"https://api-inference.huggingface.co/models/{selected_lora['repo']}"
     trigger_word = selected_lora["trigger_word"]
-    token = os.getenv("API_TOKEN")  # Make sure you have set your API token
+    token = os.getenv("API_TOKEN")
     payload = {"inputs": f"{prompt} {trigger_word}"}
     image_bytes = query(payload, api_url, token)
-    return Image.open(image_bytes)
+    return Image.open(image_bytes).resize((300, 300))  # Resizing the image
 
 # Gradio UI
 print("Before Gradio Interface")
 
 with gr.Blocks() as app:
+    title = gr.HTML("<h1>LoRA the Explorer</h1>")
+    gallery = gr.Gallery(
+        [(item["image"], item["title"]) for item in loras],
+        label="LoRA Gallery",
+        allow_preview=False,
+        columns=3,
+    )
+    prompt = gr.Textbox(label="Prompt", lines=1, max_lines=1, placeholder="Type a prompt after selecting a LoRA")
+    result = gr.Image(interactive=False, label="Generated Image")
+
     with gr.Row():
         with gr.Column():
-            title = gr.HTML("<h1>LoRA the Explorer</h1>")
-            gallery = gr.Gallery(
-                [(item["image"], item["title"]) for item in loras],
-                label="LoRA Gallery",
-                allow_preview=False,
-                columns=1,  # Change to 1 column to make images smaller
-                item_size=(150, 150)  # Set the size of the gallery images
-            )
+            title
+            gallery
         with gr.Column():
-            prompt = gr.Textbox(label="Prompt", lines=1, max_lines=1, placeholder="Type a prompt after selecting a LoRA")
+            prompt
             gr.Button("Run").click(
                 fn=run_lora,
                 inputs=[prompt],
                 outputs=[result]
             )
-            result = gr.Image(interactive=False, label="Generated Image")
+            result
 
 print("After Gradio Interface")