Update app.py

app.py

@@ -13,7 +13,7 @@ from flux.sampling import denoise, get_noise, get_schedule, prepare, unpack
 from flux.util import load_ae, load_clip, load_flow_model, load_t5
 from pulid.pipeline_flux import PuLIDPipeline
 from pulid.utils import resize_numpy_image_long
-
+from prompt_template import prompt_dict
 NSFW_THRESHOLD = 0.85
 
 def get_models(name: str, device: torch.device, offload: bool):
@@ -165,7 +165,8 @@ _HEADER_ = '''
 </div>
 
 '''  # noqa E501
-
+PROMPTS = list(prompt_dict.keys())
+DEFAULT_PROMPT_NAME = "Paris"
 
 
 def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_available() else "cpu",
@@ -176,7 +177,7 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
 
         with gr.Row():
             with gr.Column():
-                prompt = gr.Textbox(label="Prompt", value="portrait, color, cinematic")
+                prompt = gr.Dropdown(label="PROMPT", choices=PROMPTS, value=DEFAULT_PROMPT_NAME)
                 id_image = gr.Image(label="ID Image")
                 id_weight = gr.Slider(0.0, 3.0, 1, step=0.05, label="id weight")