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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -90,12 +90,15 @@ 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
  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}')
 
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
+
94
  try:
95
  language = detect(prompt)
96
 
97
  if language != 'en':
98
  prompt = GoogleTranslator(source=language, target='en').translate(prompt)
99
  print(f'\033[1mГенерация {key} перевод:\033[0m {prompt}')
100
+ except Exception as e:
101
+ prompt = prompt
102
 
103
  #prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
104
  print(f'\033[1mГенерация {key}:\033[0m {prompt}')