Update app.py

app.py

@@ -16,11 +16,11 @@ def query(payload, api_url, token):
     return io.BytesIO(response.content)
 
 # Define the function to run when the button is clicked
-def run_lora(prompt, weight):
-    selected_lora = loras[0]
+def run_lora(prompt):
+    selected_lora = loras[0]  # You can change this to select a different LoRA if needed
     api_url = f"https://api-inference.huggingface.co/models/{selected_lora['repo']}"
     trigger_word = selected_lora["trigger_word"]
-    token = os.getenv("API_TOKEN")
+    token = os.getenv("API_TOKEN")  # Make sure you have set your API token
     payload = {"inputs": f"{prompt} {trigger_word}"}
     image_bytes = query(payload, api_url, token)
     return Image.open(image_bytes)
@@ -29,32 +29,24 @@ def run_lora(prompt, weight):
 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")
-    advanced_options = gr.Accordion("Advanced options", open=False)
-    weight = gr.Slider(0, 10, value=1, step=0.1, label="LoRA weight")
-    result = gr.Image(interactive=False, label="Generated Image")
-
     with gr.Row():
         with gr.Column():
-            title
-            gallery
-            prompt
-            advanced_options
-            weight
+            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
+            )
+        with gr.Column():
+            prompt = gr.Textbox(label="Prompt", lines=1, max_lines=1, placeholder="Type a prompt after selecting a LoRA")
             gr.Button("Run").click(
                 fn=run_lora,
-                inputs=[prompt, weight],
+                inputs=[prompt],
                 outputs=[result]
             )
-        with gr.Column():
-            result
+            result = gr.Image(interactive=False, label="Generated Image")
 
 print("After Gradio Interface")