Update app.py

app.py

@@ -13,6 +13,44 @@ login(token=hf_token)
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
+SAMPLE_MODEL_IDS = [
+    'lora-library/B-LoRA-teddybear',
+    'lora-library/B-LoRA-bull',
+    'lora-library/B-LoRA-wolf_plushie',
+    'lora-library/B-LoRA-pen_sketch',
+    'lora-library/B-LoRA-cartoon_line',
+    'lora-library/B-LoRA-child',
+    'lora-library/B-LoRA-vase',
+    'lora-library/B-LoRA-scary_mug',
+    'lora-library/B-LoRA-statue',
+    'lora-library/B-LoRA-colorful_teapot',
+    'lora-library/B-LoRA-grey_sloth_plushie',
+    'lora-library/B-LoRA-teapot',
+    'lora-library/B-LoRA-backpack_dog',
+    'lora-library/B-LoRA-buddha',
+    'lora-library/B-LoRA-dog6',
+    'lora-library/B-LoRA-poop_emoji',
+    'lora-library/B-LoRA-pot',
+    'lora-library/B-LoRA-fat_bird',
+    'lora-library/B-LoRA-elephant',
+    'lora-library/B-LoRA-metal_bird',
+    'lora-library/B-LoRA-cat',
+    'lora-library/B-LoRA-dog2',
+    'lora-library/B-LoRA-drawing1',
+    'lora-library/B-LoRA-village_oil',
+    'lora-library/B-LoRA-watercolor',
+    'lora-library/B-LoRA-house_3d',
+    'lora-library/B-LoRA-ink_sketch',
+    'lora-library/B-LoRA-drawing3',
+    'lora-library/B-LoRA-crayon_drawing',
+    'lora-library/B-LoRA-kiss',
+    'lora-library/B-LoRA-drawing4',
+    'lora-library/B-LoRA-working_cartoon',
+    'lora-library/B-LoRA-painting',
+    'lora-library/B-LoRA-drawing2'
+    'lora-library/B-LoRA-multi-dog2',
+]
+
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 pipeline = StableDiffusionXLPipeline.from_pretrained(
         "stabilityai/stable-diffusion-xl-base-1.0",
@@ -124,8 +162,16 @@ with gr.Blocks(css=css) as demo:
         """)
 
         with gr.Row():    
-            content_b_lora = gr.Textbox(label="B-LoRa for content")
-            style_b_lora = gr.Textbox(label="B-LoRa for style")
+            content_b_lora = gr.Dropdown(
+                label="B-LoRa for content", 
+                allow_custom_value=True,
+                choices=SAMPLE_MODEL_IDS
+            )
+            style_b_lora = gr.Dropdown(
+                label="B-LoRa for style", 
+                allow_custom_value=True,
+                choices=SAMPLE_MODEL_IDS
+            )
         
         with gr.Column():
             load_b_loras_btn = gr.Button("load models")