Update app.py

app.py

@@ -133,20 +133,18 @@ def create_demo():
         gr.Markdown("")
         gr.Markdown("")
 
+        with gr.Column():
+            prompt_albanian = gr.Textbox(label="Përshkrimi", placeholder="Shkruani përshkrimin këtu", lines=3)
+            aspect_ratio = gr.Radio(label="Raporti i fotos", choices=list(ASPECT_RATIOS.keys()), value="1:1")
+            generate_btn = gr.Button("Gjenero")
+        
         with gr.Row():
-            with gr.Column(scale=2):
-                output_image = gr.Image(label="Imazhi i Gjeneruar", interactive=False)
-            with gr.Column(scale=1):
-                prompt_albanian = gr.Textbox(label="Përshkrimi", placeholder="Shkruani përshkrimin këtu", lines=3)
-                aspect_ratio = gr.Radio(label="Raporti i fotos", choices=list(ASPECT_RATIOS.keys()), value="1:1")
-                randomize_seed = gr.Checkbox(label="Përdor numër të rastësishëm", value=False)
-                generate_btn = gr.Button("Gjenero")
-                latency = gr.Textbox(label="Koha", interactive=False)
+            output_image = gr.Image(label="Imazhi i Gjeneruar", interactive=False)
 
         generate_btn.click(
             fn=generate_image_from_albanian,
-            inputs=[prompt_albanian, gr.Number(value=42, visible=False), aspect_ratio, randomize_seed],
-            outputs=[output_image, gr.Number(visible=False), latency],
+            inputs=[prompt_albanian, aspect_ratio],
+            outputs=[output_image],
             show_progress="full"
         )