Update app.py

app.py

@@ -3,10 +3,9 @@ import gradio as gr
 
 client = Client("kratadata/Colabor-Image-Finetune")
 
-def generate_image(prompt):
-    # Hardcode all other parameters here:
+def generate_image(prompt, aspect_ratio):
+    # Hardcode other parameters
     lora_model = "group1"
-    aspect_ratio = "1:1"
     num_inference_steps = 28
     guidance_scale = 3
     seed = -1
@@ -27,8 +26,9 @@ def generate_image(prompt):
 with gr.Blocks(css=".svelte-1ipelgc { display: none !important; }") as iface:
     gr.Markdown("")
     prompt = gr.Textbox(label="Përshkrimi", placeholder="Shkruaj përshkrimin këtu...")
-    output = gr.Image(label="Rezultati")
+    aspect = gr.Dropdown(["1:1", "16:9", "9:16"], label="Dimensionet e fotos", value="1:1")
     btn = gr.Button("Gjenero")
-    btn.click(fn=generate_image, inputs=prompt, outputs=output)
+    output = gr.Image(label="Rezultati")
+    btn.click(fn=generate_image, inputs=[prompt, aspect], outputs=output)
 
 iface.launch(share=True)