Update app.py

app.py

@@ -60,6 +60,7 @@ def infer(
         generator=generator,
         use_resolution_binning=use_resolution_binning,
     ).images[0]
+    image.save("output_image.png")
     return image, seed
 
 examples = [