Update app.py

app.py

@@ -71,6 +71,8 @@ def generate_image_from_text(prompt: str) -> None:
 
         # Display the image inline for Jupyter Notebook or applications with inline display support
         display.display(image)
+        image.save("generated_image.png")
+        print("Image saved as generated_image.png. Open the file to view it.")
 
     except Exception as e:
         print(f"Error generating image: {str(e)}")