Update app.py

app.py

@@ -57,6 +57,11 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
             try:
                 # Пытаемся извлечь текст из ответа
                 prompt = response_json["choices"][0]["message"]["content"]
+            except Exception as e:
+                print(f"Error processing the image response: {e}")
+        else:
+            # Если произошла ошибка, возвращаем сообщение об ошибке
+            print(f"Error: {response.status_code} - {response.text}")
         
     
     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