Update app.py

app.py

@@ -10,6 +10,9 @@ from datasets import load_dataset
 dataset = load_dataset("Gustavosta/Stable-Diffusion-Prompts")
 prompt_df = dataset["train"].to_pandas()
 
+DEFAULT_MODEL = "stabilityai/stable-diffusion-2-1"
+DEFAULT_PROMPT = "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"
+
 def get_samples():
     prompt_list = prompt_df.sample(n = 10)["Prompt"].map(lambda x: x).values.tolist()
     return prompt_list
@@ -67,7 +70,9 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
     API_TOKEN = os.environ.get("HF_READ_TOKEN")
     headers = {"Authorization": f"Bearer {API_TOKEN}"}
     if type(current_model) != type(""):
-        current_model = "stabilityai/stable-diffusion-2-1"
+        current_model = DEFAULT_MODEL
+    if type(prompt) != type(""):
+        prompt = DEFAULT_PROMPT
     api = InferenceApi(current_model)
     api.headers = headers
 
@@ -225,15 +230,17 @@ with gr.Blocks(css=css) as demo:
         with gr.Column():
             btn_refresh = gr.Button(value="Click Refresh to get current deployed models")
             btn_refresh.click(None, js="window.location.reload()")
-            current_model = gr.Dropdown(label="Current Model", choices=list_models, value="stabilityai/stable-diffusion-2-1")
+            current_model = gr.Dropdown(label="Current Model", choices=list_models, value=DEFAULT_MODEL)
         
     with gr.Row("prompt-container"):
         text_prompt = gr.Textbox(label="Prompt", placeholder="a cute dog", 
-                                     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",
+                                     value = DEFAULT_PROMPT,
                                      lines=2, elem_id="prompt-text-input")
         text_button = gr.Button("Manualy input Generate", variant='primary', elem_id="gen-button")
     with gr.Row("prompt-container"):
-        select_prompt = gr.Dropdown(label="prompt selected", choices=list_prompts, value=list_prompts[1])
+        select_prompt = gr.Dropdown(label="prompt selected", choices=list_prompts, 
+                                   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"
+                                   )
         select_button = gr.Button("Select Prompt Generate", variant='primary', elem_id="gen-button")
         
     with gr.Row():