Update main.py

main.py

@@ -29,7 +29,7 @@ def generate_image(prompt):
         return image
     except Exception as e:
         # Handle the error gracefully, such as logging the error or returning a default image
-        print(f"Error generating image: {e}")
+        raise gr.Error(f"Error generating image: {e}")
         return "Experiencing high demand. Please retry shortly. Thank you for your patience"  # Return None or a default image in case of error