Update app.py

app.py

@@ -46,17 +46,20 @@ except Exception as e:
     exit()  # Stop if model loading fails
 
 # 4. Image generation function (now decorated)
-@GPU(duration=65)  # ONLY if in a HF Space, remove if not
+@GPU(duration=65)  # Only if in HF Space
 def generate_image(prompt):
     global pipeline
-    if pipeline is None:  # Should not happen, but good to check
+    if pipeline is None:
         return "Error: Model not loaded!"
 
     try:
-        image = pipeline(prompt).images[0]  # Access the first image from the list
-        return image
+        image = pipeline(prompt).images[0]  # Try to generate the image
+        print("Image generated successfully!") # Print success message (for debugging)
+        return image # Return the image if successful
     except Exception as e:
-        return f"Error generating image: {e}"
+        error_message = f"Error during image generation: {e}" # Capture error
+        print(error_message) # Print error message to console
+        return error_message  # Return the error message to Gradio (so it shows up)
 
 # 5. Gradio interface
 with gr.Blocks() as demo: