Update app.py

app.py

@@ -28,5 +28,7 @@ def pil_to_base64_string(pil_image):
 # Launch the interface
 iface.launch()
 
-# After the interface is launched, you can download the output image as follows:
-output_pil = iface.get_output() # get the output PIL image from the
+output_pil = iface.get_output() # get the output PIL image from the interface
+output_base64_str = pil_to_base64_string(output_pil) # convert the PIL image to base64 string
+with open("output_image.png", "wb") as f:
+    f.write(base64.b64decode(output_base64_str)) # write the decoded base64 string to a file