Update app.py

app.py

@@ -104,8 +104,8 @@ def generate(
             num_inference_steps=num_inference_steps_base,
             generator=generator,
             output_type="pil",
-        ).images[0]
-        return Image
+        ).images
+        images[0].save("fantasy_landscape.png")
 
 examples = [
     "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",