Update app.py

app.py

@@ -17,24 +17,6 @@ def get_samples():
     prompt_list = prompt_df.sample(n = 10)["Prompt"].map(lambda x: x).values.tolist()
     return prompt_list
 
-'''
-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 gr.Dropdown(label="Current Model", choices=list_models, value=DEFAULT_MODEL,
-                                       info = "default model: {}".format(DEFAULT_MODEL)
-                                       ), gr.Dropdown(label="Prompt selected", choices=get_samples(), 
-                                   value = "1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt",
-                                   info = "default prompt: {}".format(DEFAULT_PROMPT)
-                                   )
-    #return list_models, get_samples()
-'''
-
 def update_models():
     client = InferenceClient()
     models = client.list_deployed_models()
@@ -43,7 +25,14 @@ def update_models():
 
 def update_prompts():
     return gr.Dropdown.update(choices=get_samples())
-    
+
+def get_params(request: gr.Request):
+    params = request.query_params
+    ip = request.client.host
+    req = {"params": params,
+          "ip": ip}
+    return update_models(), update_prompts()
+
 client = InferenceClient()
 models = client.list_deployed_models()
 list_models = models["text-to-image"]
@@ -286,6 +275,6 @@ with gr.Blocks(css=css) as demo:
     btn_refresh.click(update_models, None, current_model)
     btn_refresh.click(update_prompts, None, select_prompt)
     
-    #demo.load(get_params, None, [current_model, select_prompt])
+    demo.load(get_params, None, [current_model, select_prompt])
 
 demo.launch(show_api=False)