Rooni commited on
Commit
7bc9976
·
verified ·
1 Parent(s): 3899612

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -90,11 +90,12 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
90
  print(f"Error: {response.status_code} - {response.text}")
91
  API_TOKEN = random.choice([os.getenv("HF_READ_TOKEN"), os.getenv("HF_READ_TOKEN_2"), os.getenv("HF_READ_TOKEN_3"), os.getenv("HF_READ_TOKEN_4"), os.getenv("HF_READ_TOKEN_5")]) # it is free
92
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
93
- language = detect(prompt)
 
94
 
95
- if language != 'en':
96
- prompt = GoogleTranslator(source=language, target='en').translate(prompt)
97
- print(f'\033[1mГенерация {key} перевод:\033[0m {prompt}')
98
 
99
  #prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
100
  print(f'\033[1mГенерация {key}:\033[0m {prompt}')
 
90
  print(f"Error: {response.status_code} - {response.text}")
91
  API_TOKEN = random.choice([os.getenv("HF_READ_TOKEN"), os.getenv("HF_READ_TOKEN_2"), os.getenv("HF_READ_TOKEN_3"), os.getenv("HF_READ_TOKEN_4"), os.getenv("HF_READ_TOKEN_5")]) # it is free
92
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
93
+ try:
94
+ language = detect(prompt)
95
 
96
+ if language != 'en':
97
+ prompt = GoogleTranslator(source=language, target='en').translate(prompt)
98
+ print(f'\033[1mГенерация {key} перевод:\033[0m {prompt}')
99
 
100
  #prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
101
  print(f'\033[1mГенерация {key}:\033[0m {prompt}')