Update app.py

app.py

@@ -108,10 +108,10 @@ def generate(
 
     if not apply_refiner:
         return pipe(
-            prompt=prompt,
-            negative_prompt=negative_prompt,
-            prompt_2=prompt_2,
-            negative_prompt_2=negative_prompt_2,
+            prompt=process_text(prompt),
+            negative_prompt=process_text(negative_prompt),
+            prompt_2=process_text(prompt_2),
+            negative_prompt_2=process_text(negative_prompt_2),
             width=width,
             height=height,
             guidance_scale=guidance_scale_base,
@@ -121,10 +121,10 @@ def generate(
         ).images[0]
     else:
         latents = pipe(
-            prompt=prompt,
-            negative_prompt=negative_prompt,
-            prompt_2=prompt_2,
-            negative_prompt_2=negative_prompt_2,
+            prompt=process_text(prompt),
+            negative_prompt=process_text(negative_prompt),
+            prompt_2=process_text(prompt_2),
+            negative_prompt_2=process_text(negative_prompt_2),
             width=width,
             height=height,
             guidance_scale=guidance_scale_base,
@@ -133,10 +133,10 @@ def generate(
             output_type="latent",
         ).images
         image = refiner(
-            prompt=prompt,
-            negative_prompt=negative_prompt,
-            prompt_2=prompt_2,
-            negative_prompt_2=negative_prompt_2,
+            prompt=process_text(prompt),
+            negative_prompt=process_text(negative_prompt),
+            prompt_2=process_text(prompt_2),
+            negative_prompt_2=process_text(negative_prompt_2),
             guidance_scale=guidance_scale_refiner,
             num_inference_steps=num_inference_steps_refiner,
             image=latents,
@@ -159,13 +159,13 @@ with gr.Blocks(css="style.css") as demo:
     )
     with gr.Group():
         with gr.Row():
-            prompt = process_text(gr.Text(
+            prompt = gr.Text(
                 label="提示词",
                 show_label=False,
                 max_lines=1,
                 placeholder="输入要生成的画面内容",
                 container=False,
-            ))
+            )
             run_button = gr.Button("生成", scale=0)
         result = gr.Image(label="生成结果", show_label=False)
     with gr.Accordion("高级选项", open=False):
@@ -173,24 +173,24 @@ with gr.Blocks(css="style.css") as demo:
             use_negative_prompt = gr.Checkbox(label="使用反向提示词", value=False)
             use_prompt_2 = gr.Checkbox(label="使用提示词 2", value=False)
             use_negative_prompt_2 = gr.Checkbox(label="使用反向提示词 2", value=False)
-        negative_prompt = process_text(gr.Text(
+        negative_prompt = gr.Text(
             label="反向提示词",
             max_lines=1,
             placeholder="输入不想在画面中出现的内容，比如：“胡子”，“人群”",
             visible=False,
-        ))
-        prompt_2 = process_text(gr.Text(
+        )
+        prompt_2 = gr.Text(
             label="提示词 2",
             max_lines=1,
             placeholder="输入你的提示词",
             visible=False,
-        ))
-        negative_prompt_2 = process_text(gr.Text(
+        )
+        negative_prompt_2 = gr.Text(
             label="反向提示词 2",
             max_lines=1,
             placeholder="输入你的反向提示词",
             visible=False,
-        ))
+        )
 
         seed = gr.Slider(
             label="种子数",