Update app.py

app.py

@@ -13,11 +13,16 @@ def download_image(output_image):
     return gradio.Interface.download(img_bytes, "output_image.png")
 
 # define input and output interfaces
-input_image = gr.inputs.Image(type="file", label="Input Image")
-output_image = gr.outputs.Image(type="file", label="Output Image")
+input_image = gr.inputs.Image(type="filepath", label="Input Image")
+output_image = gr.outputs.Image(type="filepath", label="Output Image")
 
 # define the download button
-download_button = gr.InterfaceFn(download_image, output_image, title="Download")
+download_button = gr.Interface(
+    fn=download_image,
+    inputs=output_image,
+    outputs="download",
+    title="Download"
+)
 
 # create the interface
 iface = gr.Interface(