Update app.py

app.py

@@ -28,8 +28,8 @@ img_to_text = gr.Blocks.load(name="spaces/pharma/CLIP-Interrogator")
 language_translation_model = hub.Module(name='baidu_translate')
 language_recognition_model = hub.Module(name='baidu_language_recognition')
 
-style_list = ['古风', '油画', '水彩', '卡通', '二次元', '浮世绘', '蒸汽波艺术', 'low poly', '像素风格', '概念艺术', '未来主义', '赛博朋克', '写实风格', '洛丽塔风格', '巴洛克风格', '超现实主义', '探索无限']
-style_list_EN = ['Chinese Ancient Style', 'Oil painting', 'Watercolor', 'Cartoon', 'Anime', 'Ukiyoe', 'Vaporwave', 'low poly', 'Pixel Style', 'Conceptual Art', 'Futurism', 'Cyberpunk', 'Realistic style', 'Lolita style', 'Baroque style', 'Surrealism', '']
+style_list = ['古风', '油画', '水彩', '卡通', '二次元', '浮世绘', '蒸汽波艺术', 'low poly', '像素风格', '概念艺术', '未来主义', '赛博朋克', '写实风格', '洛丽塔风格', '巴洛克风格', '超现实主义', '默认']
+style_list_EN = ['Chinese Ancient Style', 'Oil painting', 'Watercolor', 'Cartoon', 'Anime', 'Ukiyoe', 'Vaporwave', 'low poly', 'Pixel Style', 'Conceptual Art', 'Futurism', 'Cyberpunk', 'Realistic style', 'Lolita style', 'Baroque style', 'Surrealism', 'Default']
 
 tips = {"en": "Tips: The input text will be translated into Chinese for generation", 
         "jp": "ヒント: 入力テキストは生成のために中国語に翻訳されます", 
@@ -55,8 +55,8 @@ def translate_language(text_prompts):
         count += 1
         tips_text = None
         language_code = language_recognition_model.recognize(text_prompts)
-        if language_code != 'zh':
-            text_prompts = language_translation_model.translate(text_prompts, language_code, 'zh')
+        if language_code != 'en':
+            text_prompts = language_translation_model.translate(text_prompts, language_code, 'en')
     except Exception as e:
         error_text = str(e)
         return {status_text:error_text, language_tips_text:gr.update(visible=False)}
@@ -119,7 +119,7 @@ def merge_video(music, image):
     fps = 12
     slide_time = audio_length
     fourcc = cv2.VideoWriter.fourcc(*'MJPG')
-    out = cv2.VideoWriter(file_name, fourcc, fps, (1024, 1024))
+    out = cv2.VideoWriter(file_name, fourcc, fps, (512, 512))
     
     # for image in img_list:
     #     cv_img = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)