Update app.py

app.py

@@ -41,9 +41,8 @@ def extend_prompt(input_text):
  return extended_prompt
 
 def generate_image(prompt, selected_model, is_negative=False, steps=1, cfg_scale=6, seed=None):
- if not input_text.strip():
- gr.Warning("Input text is empty!")
- return None
+ if not prompt.strip():
+ raise gr.Error("Cannot generate image: Input text is empty!")
  
  model_url = select_model(selected_model)
  API_URL = f"{model_url}"