Update app.py

app.py

@@ -36,10 +36,10 @@ def extend_prompt(input_text):
     payload = {"inputs": input_text}
 
     try:
-        response = requests.post(API_URL, headers=HEADERS, json=payload).json()
+        response = requests.post(API_URL, headers=HEADERS, json=payload).raise_for_status().json()
         return response[0].get("generated_text", "")
     except requests.exceptions.RequestException as e:
-        gr.Error(f"Error in API request: {e}")
+        gr.Warning(f"Error in API request: {e}")
         return None
 
 def generate_image(prompt, selected_model, is_negative=False, steps=1, cfg_scale=6, seed=None):