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}"