Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from huggingface_hub import login, HfFileSystem
+from huggingface_hub import login, HfFileSystem, HfApi
 
 from diffusers import DiffusionPipeline, StableDiffusionXLPipeline
 import torch
@@ -13,6 +13,7 @@ hf_token = os.environ.get("HF_TOKEN")
 login(token = hf_token)
 
 fs = HfFileSystem(token=hf_token)
+api = HfApi()
 
 original_pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
 
@@ -160,11 +161,25 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             
             with gr.Column():
+
+                if is_shared_ui:
+                    
+                    your_username = api.whoami()["name"]
+                    my_models = api.list_models(author=your_username, filter="diffusers")
+                    print(my_models)
                 
-                lora_1_id = gr.Textbox(
-                    label = "LoRa 1 ID",
-                    placeholder = "username/model_id"
-                )
+                    lora_1_id = gr.Dropdown()(
+                        label = "LoRa 1 ID",
+                        choices = my_models,
+                        allow_custom_value = True
+                        #placeholder = "username/model_id"
+                    )
+                
+                else:
+                    lora_1_id = gr.Textbox(
+                        label = "LoRa 1 ID",
+                        placeholder = "username/model_id"
+                    )
 
                 lora_1_sfts = gr.Dropdown(
                     label = "Safetensors file",