Update app.py

app.py

@@ -81,7 +81,6 @@ def process_text(prompt):
         print("prompt: \n", prompt_trans)
         return prompt_trans
 
-process_text("外面做一个测试")
 
 @spaces.GPU
 def generate(
@@ -110,16 +109,16 @@ def generate(
     if not use_negative_prompt_2:
         negative_prompt_2 = None  # type: ignore
 
-    process_text("里面做一个测试")
-    print("prompt是：", prompt)
-    print("negative_prompt是：", negative_prompt)
-    print("prompt_2是：", prompt_2)
-    print("negative_prompt_2是：", negative_prompt_2)
+    # process_text("里面做一个测试")
+    # print("prompt是：", prompt)
+    # print("negative_prompt是：", negative_prompt)
+    # print("prompt_2是：", prompt_2)
+    # print("negative_prompt_2是：", negative_prompt_2)
     
     if not apply_refiner:
         return pipe(
-            prompt=prompt,
-            negative_prompt=negative_prompt,
+            prompt=process_text(prompt),
+            negative_prompt=process_text(negative_prompt),
             prompt_2=prompt_2,
             negative_prompt_2=negative_prompt_2,
             width=width,
@@ -131,8 +130,8 @@ def generate(
         ).images[0]
     else:
         latents = pipe(
-            prompt=prompt,
-            negative_prompt=negative_prompt,
+            prompt=process_text(prompt),
+            negative_prompt=process_text(negative_prompt),
             prompt_2=prompt_2,
             negative_prompt_2=negative_prompt_2,
             width=width,
@@ -143,8 +142,8 @@ def generate(
             output_type="latent",
         ).images
         image = refiner(
-            prompt=prompt,
-            negative_prompt=negative_prompt,
+            prompt=process_text(prompt),
+            negative_prompt=process_text(negative_prompt),
             prompt_2=prompt_2,
             negative_prompt_2=negative_prompt_2,
             guidance_scale=guidance_scale_refiner,