Update app.py

app.py

@@ -5,6 +5,14 @@ import random
 import os
 from PIL import Image
 from huggingface_hub import InferenceApi, InferenceClient
+from datasets import load_dataset
+
+dataset = load_dataset("Gustavosta/Stable-Diffusion-Prompts")
+prompt_df = dataset["train"].to_pandas()
+
+def get_samples():
+    prompt_list = prompt_df.sample(n = 10)["Prompt"].values.tolist()
+    return prompt_list
 
 def get_params(request: gr.Request):
     params = request.query_params
@@ -14,7 +22,7 @@ def get_params(request: gr.Request):
     client = InferenceClient()
     models = client.list_deployed_models()
     list_models = models["text-to-image"]
-    return list_models
+    return list_models, get_samples()
     
 client = InferenceClient()
 models = client.list_deployed_models()
@@ -232,20 +240,16 @@ with gr.Blocks(css=css) as demo:
         image_style = gr.Dropdown(label="Style", choices=["None style", "Cinematic", "Digital Art", "Portrait"], value="Portrait", allow_custom_value=False)
 
     with gr.Row():
-        gr.Examples(
-            [
-            "Chinese ink painting",
-            "silk road",
-            "masterpiece, best quality, 1girl, solo, crop top, denim shorts, choker, (graffiti:1.5), paint splatter, arms behind back, against wall, looking at viewer, armband, thigh strap, paint on body, head tilt, bored, multicolored hair, aqua eyes, headset,",
-            "beautiful home",
-            "interior design of living room",
-            "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",
-            ],
-            inputs = text_prompt,
-            label = "Prompt Examples"
-        )
+        with gr.Columns():
+            exps = gr.Examples(
+                get_samples(),
+                inputs = text_prompt,
+                label = "Prompt Examples"
+            )
+            exp_refresh = gr.Button(value="Click Refresh to get newly prompt examples")
+            exp_refresh.click(get_samples, None, exps)
     
     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.load(get_params, None, [current_model, exps])
 
 demo.launch(show_api=False)