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):