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],