Update app.py

app.py

@@ -3,9 +3,7 @@ import gradio as gr
 
 client = Client("kratadata/Colabor-Image-Finetune")
 
-def generate_image(prompt, aspect_ratio):
-    # Change lora_model to a valid choice
-    lora_model = "pizza-ananas"  # or "pizza-pilz" or "pizza-ananas-pilz"
+def generate_image(prompt, aspect_ratio, lora_model):
     num_inference_steps = 28
     guidance_scale = 3
     seed = -1
@@ -24,11 +22,17 @@ def generate_image(prompt, aspect_ratio):
     return result
 
 with gr.Blocks(css=".svelte-1ipelgc { display: none !important; }") as iface:
-    gr.Markdown("")
+    gr.Markdown("# Gjeneruesi i Imazheve me Model të Personalizuar")
+    
     prompt = gr.Textbox(label="Përshkrimi", placeholder="Shkruaj përshkrimin këtu...")
     aspect = gr.Dropdown(["1:1", "16:9", "9:16"], label="Dimensionet e fotos", value="1:1")
+    model = gr.Dropdown(
+        choices=["pizza-ananas", "pizza-pilz", "pizza-ananas-pilz"],
+        label="Zgjidh Modelin e Imazhit"
+    )
     btn = gr.Button("Gjenero")
     output = gr.Image(label="Rezultati")
-    btn.click(fn=generate_image, inputs=[prompt, aspect], outputs=output)
+
+    btn.click(fn=generate_image, inputs=[prompt, aspect, model], outputs=output)
 
 iface.launch(share=True)