DGSpitzer commited on
Commit
2354620
1 Parent(s): 130e9b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -29,8 +29,8 @@ text_to_music = gr.Interface.load("spaces/fffiloni/text-2-music")
29
  language_translation_model = hub.Module(name='baidu_translate')
30
  language_recognition_model = hub.Module(name='baidu_language_recognition')
31
 
32
- style_list = ['古风', '油画', '水彩', '卡通', '二次元', '浮世绘', '蒸汽波艺术', 'low poly', '像素风格', '概念艺术', '未来主义', '赛博朋克', '写实风格', '洛丽塔风格', '巴洛克风格', '超现实主义', '默认']
33
- 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']
34
 
35
  tips = {"en": "Tips: The input text will be translated into English for generation",
36
  "jp": "ヒント: 入力テキストは生成のために中国語に翻訳されます",
@@ -73,7 +73,7 @@ def translate_language(text_prompts):
73
 
74
 
75
  def get_result(text_prompts, style_indx, musicAI_indx):
76
- style = style_list[style_indx]
77
  prompt = style + "," + text_prompts
78
 
79
  sdresult = pipe(prompt)
 
29
  language_translation_model = hub.Module(name='baidu_translate')
30
  language_recognition_model = hub.Module(name='baidu_language_recognition')
31
 
32
+ # style_list = ['古风', '油画', '水彩', '卡通', '二次元', '浮世绘', '蒸汽波艺术', 'low poly', '像素风格', '概念艺术', '未来主义', '赛博朋克', '写实风格', '洛丽塔风格', '巴洛克风格', '超现实主义', '默认']
33
+ 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', 'Detailed']
34
 
35
  tips = {"en": "Tips: The input text will be translated into English for generation",
36
  "jp": "ヒント: 入力テキストは生成のために中国語に翻訳されます",
 
73
 
74
 
75
  def get_result(text_prompts, style_indx, musicAI_indx):
76
+ style = style_list_EN[style_indx]
77
  prompt = style + "," + text_prompts
78
 
79
  sdresult = pipe(prompt)