Update app.py

app.py

@@ -116,7 +116,7 @@ def load_img(filename, norm=True):
 
 
 def inference(image, upscale, large_input_flag, color_fix):
-	if upscale is None or not isinstance(upscale, (int, float)) or upscale == 3:
+	if upscale is None or not isinstance(upscale, (int, float)) or upscale == 3.:
 		upscale = 2
 		
 	upscale = int(upscale) # convert type to int
@@ -165,14 +165,12 @@ 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)
 	
-    return img, Image.fromarray(output)
-
 	# # save restored img
 	# save_path = f'results/out.png'
 	# cv2.imwrite(save_path, output)
 
 	# output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
-	# return output, save_path
+	return Image.fromarray(output), img