yizhangliu commited on
Commit
91ad1e7
·
1 Parent(s): f34eb4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -2
app.py CHANGED
@@ -6,6 +6,15 @@ from loguru import logger
6
 
7
  language_translation_model = hub.Module(directory=f'./baidu_translate')
8
  def getTextTrans(text, source='zh', target='en'):
 
 
 
 
 
 
 
 
 
9
  try:
10
  text_translation = language_translation_model.translate(text, source, target)
11
  return text_translation
@@ -159,7 +168,7 @@ start_work = """async() => {
159
  return false;
160
  }"""
161
 
162
- def prompt_extend(prompt):
163
  prompt_en = getTextTrans(prompt, source='zh', target='en')
164
  extend_prompt_en = extend_prompt_pipe(prompt_en+',', num_return_sequences=1)[0]["generated_text"]
165
  if (prompt != prompt_en):
@@ -210,4 +219,4 @@ with gr.Blocks(title='Text-to-Image') as demo:
210
 
211
  # submit_btn.click(fn=extend_prompt, inputs=[prompt_input0], outputs=[prompt_input0_en, prompt_input1, prompt_input2])
212
 
213
- demo.launch()
 
6
 
7
  language_translation_model = hub.Module(directory=f'./baidu_translate')
8
  def getTextTrans(text, source='zh', target='en'):
9
+ def is_chinese(string):
10
+ for ch in string:
11
+ if u'\u4e00' <= ch <= u'\u9fff':
12
+ return True
13
+ return False
14
+
15
+ if not is_chinese(text):
16
+ return text
17
+
18
  try:
19
  text_translation = language_translation_model.translate(text, source, target)
20
  return text_translation
 
168
  return false;
169
  }"""
170
 
171
+ def prompt_extend(prompt):
172
  prompt_en = getTextTrans(prompt, source='zh', target='en')
173
  extend_prompt_en = extend_prompt_pipe(prompt_en+',', num_return_sequences=1)[0]["generated_text"]
174
  if (prompt != prompt_en):
 
219
 
220
  # submit_btn.click(fn=extend_prompt, inputs=[prompt_input0], outputs=[prompt_input0_en, prompt_input1, prompt_input2])
221
 
222
+ demo.launch()