Update app.py

app.py

@@ -36,15 +36,24 @@ class calculateDuration:
             print(f"Elapsed time: {self.elapsed_time:.6f} seconds")
 
 
-def update_selection(evt: gr.SelectData):
+def update_selection(evt: gr.SelectData, width, height):
     selected_lora = loras[evt.index]
     new_placeholder = f"Type a prompt for {selected_lora['title']}"
     lora_repo = selected_lora["repo"]
     updated_text = f"### Selected: [{lora_repo}](https://huggingface.co/{lora_repo}) ✨"
+    if "aspect" in selected_lora:
+        if selected_lora["aspect"] == "portrait":
+            width = 768
+            height = 1024
+        elif selected_lora["aspect"] == "landscape":
+            width = 1024
+            height = 768
     return (
         gr.update(placeholder=new_placeholder),
         updated_text,
-        evt.index
+        evt.index,
+        width,
+        height,
     )
 
 @spaces.GPU(duration=70)
@@ -138,7 +147,11 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
                     seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True)
                     lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=1, step=0.01, value=0.95)
 
-    gallery.select(update_selection, outputs=[prompt, selected_info, selected_index])
+    gallery.select(
+        update_selection,
+        inputs=[width, height]
+        outputs=[prompt, selected_info, selected_index, width, height]
+    )
 
     gr.on(
         triggers=[generate_button.click, prompt.submit],