Update app.py

app.py

@@ -14,15 +14,17 @@ def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Ka
     is_api_call = huggingface_api_key is not None
 
     if is_api_call:
+        
+        # Use the environment variable for the API key in GUI mode
+        API_TOKEN = os.getenv("HF_READ_TOKEN")
+        headers = {"Authorization": f"Bearer {API_TOKEN}"}
+        
+    else:
         # Validate the API key if it's an API call
         if huggingface_api_key == "":
             raise gr.Error("API key is required for API calls.")
         
         headers = {"Authorization": f"Bearer {huggingface_api_key}"}
-    else:
-        # Use the environment variable for the API key in GUI mode
-        API_TOKEN = os.getenv("HF_READ_TOKEN")
-        headers = {"Authorization": f"Bearer {API_TOKEN}"}
     
     if prompt == "" or prompt is None:
         return None