Spaces:
Running
Running
Commit
·
91ad1e7
1
Parent(s):
f34eb4b
Update app.py
Browse files
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()
|