Update app.py

app.py

@@ -152,11 +152,11 @@ def inference(image, upscale, large_input_flag, color_fix):
 		output = np.transpose(output[[2, 1, 0], :, :], (1, 2, 0))
 	output = (output * 255.0).round().astype(np.uint8)
 	
-    # # save results
-	# save_path = './out.png'
-	# cv2.imwrite(save_path, output)
+    # save results
+	save_path = './out.png'
+	cv2.imwrite(save_path, output)
 	
-	return (image, Image.fromarray(output)), output #, save_path
+	return (image, Image.fromarray(output)), save_path
     
 
 
@@ -239,13 +239,13 @@ demo = gr.Interface(
                         type="pil",
                         show_download_button=True,
                     ),
-		# gr.Image(
-		# 	label="Download Output",
-		# 	type='filepath'
-		# ),
 		gr.Image(
-			label="Download Output"
+			label="Download Output",
+			type='filepath',
 		),
+		# gr.Image(
+		# 	label="Download Output"
+		# ),
     ],
     title=title,
     description=description,