Update app.py

app.py

@@ -167,7 +167,9 @@ def real_img2img_to_anime(text, input_img, chinese_check):
     input_img = Image.fromarray(input_img)
     input_img = load_image(input_img)
     compel_proc = Compel(tokenizer=pipe_scribble.tokenizer, text_encoder=pipe_scribble.text_encoder)
-    if chinese_check:
+    lang_check_label = pipe(text, top_k=1, truncation=True)[0]['label']
+    lang_check_score = pipe(text, top_k=1, truncation=True)[0]['score']
+    if lang_check_label == 'zh' and lang_check_score >= 0.85:
         text = translate(text)
         print("prompt text:", text)
     prompt = text + hidden_booster_text