Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -7,6 +7,8 @@ import os
|
|
7 |
from io import BytesIO
|
8 |
import html
|
9 |
import re
|
|
|
|
|
10 |
|
11 |
|
12 |
|
@@ -157,6 +159,12 @@ for model_name in model_list:
|
|
157 |
|
158 |
|
159 |
def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
|
|
|
|
|
|
|
|
|
|
|
|
|
160 |
result = prodia_client.generate({
|
161 |
"prompt": prompt,
|
162 |
"negative_prompt": negative_prompt,
|
|
|
7 |
from io import BytesIO
|
8 |
import html
|
9 |
import re
|
10 |
+
from deep_translator import GoogleTranslator
|
11 |
+
from langdetect import detect
|
12 |
|
13 |
|
14 |
|
|
|
159 |
|
160 |
|
161 |
def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
|
162 |
+
language = detect(prompt)
|
163 |
+
|
164 |
+
if language == 'ru':
|
165 |
+
prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
|
166 |
+
print(prompt)
|
167 |
+
|
168 |
result = prodia_client.generate({
|
169 |
"prompt": prompt,
|
170 |
"negative_prompt": negative_prompt,
|