Update app.py

app.py

@@ -27,6 +27,7 @@ mmodels = {
     "CleanLinearMix": "https://api-inference.huggingface.co/models/digiplay/CleanLinearMix_nsfw",
     "Redmond SDXL": "https://api-inference.huggingface.co/models/artificialguybr/LogoRedmond-LogoLoraForSDXL-V2"
 }
+API_TOKEN = os.getenv("HF_READ_TOKEN")
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 timeout = 100
 # PLEASE ❤ like ❤ this space. Please like me. I am 12 years old, one of my projects is: https://ai-hub.rf.gd . I live in Russia, I don't know English very well. Therefore, I apologize that there is only Russian here, but I think it will not be difficult to translate all this. (For example, using gpt)
@@ -53,7 +54,7 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
             "max_tokens": 4095,
         }    
 
-
+     
         # API ключ для OpenAI
         api_key_oi = os.getenv("API_KEY_OPENAI")
 
@@ -81,11 +82,10 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
         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
-    headers = {"Authorization": f"Bearer {API_TOKEN}"}
-    
+    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")])
+    headers = {
+            'Authorization': f'Bearer {API_TOKEN}'
+    }
     prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
     print(f'\033[1mГенерация {key} перевод:\033[0m {prompt}')