Update app.py

app.py

@@ -6,8 +6,17 @@ import os
 from PIL import Image
 from huggingface_hub import InferenceApi, InferenceClient
 
+def get_params(request: gr.Request):
+    params = request.query_params
+    ip = request.client.host
+    req = {"params": params,
+          "ip": ip}
+    client = InferenceClient()
+    models = client.list_deployed_models()
+    list_models = models["text-to-image"]
+    return list_models
+    
 client = InferenceClient()
-
 models = client.list_deployed_models()
 list_models = models["text-to-image"]
 
@@ -202,7 +211,9 @@ with gr.Blocks(css=css) as demo:
     )
     
     with gr.Row(elem_id="prompt-container"):
-        current_model = gr.Dropdown(label="Current Model", choices=list_models, value=list_models[1])
+        btn_refresh = gr.Button(value="Refresh to get deployed models")
+        btn_refresh.click(None, _js="window.location.reload()")
+        current_model = gr.Dropdown(label="Current Model", choices=list_models, value=list_models[0])
         
     with gr.Row(elem_id="prompt-container"):
         text_prompt = gr.Textbox(label="Prompt", placeholder="a cute dog", lines=1, elem_id="prompt-text-input")
@@ -216,5 +227,6 @@ with gr.Blocks(css=css) as demo:
         image_style = gr.Dropdown(label="Style", choices=["None style", "Cinematic", "Digital Art", "Portrait"], value="None style", allow_custom_value=False)
 
     text_button.click(generate_txt2img, inputs=[current_model, text_prompt, negative_prompt, image_style], outputs=image_output)
+    demo.load(get_params, None, current_model)
 
 demo.launch(show_api=False)