Update app.py

app.py

@@ -27,13 +27,12 @@ def query(prompt, model, custom_lora, is_negative=False, steps=35, cfg_scale=7,
     prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
     print(f'Generation {key}: {prompt}')
 
+    # Set API URL based on model selection
     try:
-        response = None  # Initialize response variable
+        if custom_lora.strip():
+            API_URL = f"https://api-inference.huggingface.co/models/{custom_lora.strip()}"
+        else:
 
-    # Set API URL based on model selection
-    if custom_lora.strip():
-        API_URL = f"https://api-inference.huggingface.co/models/{custom_lora.strip()}"
-    else:
         if model == 'Stable Diffusion XL':
             API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
         if model == 'FLUX.1 [Dev]':
@@ -257,12 +256,12 @@ def query(prompt, model, custom_lora, is_negative=False, steps=35, cfg_scale=7,
         
     except requests.exceptions.RequestException as e:
         error_message = f"Request failed: {str(e)}"
-        if response:
-            if response.status_code == 401:
+        if hasattr(e, 'response') and e.response is not None:
+            if e.response.status_code == 401:
                 error_message = "Invalid API token. Please check your Hugging Face API token."
-            elif response.status_code == 403:
+            elif e.response.status_code == 403:
                 error_message = "Access denied. Please check your API token permissions."
-            elif response.status_code == 503:
+            elif e.response.status_code == 503:
                 error_message = "Model is currently loading. Please try again in a few moments."
         raise gr.Error(error_message)
     except Exception as e: