Update app.py

app.py

@@ -111,13 +111,12 @@ def inference(image, upscale, large_input_flag, color_fix):
 	upscale = int(upscale) # convert type to int
 
 	model = set_safmn(upscale)
-	print(image)
 
-	img = cv2.imread(str(image), cv2.IMREAD_COLOR)
-	print(f'input size: {img.shape}')
+	# img = cv2.imread(str(image), cv2.IMREAD_COLOR)
+	# print(f'input size: {img.shape}')
 
 	# img2tensor
-	y = y.astype(np.float32) / 255.
+	y = image.astype(np.float32) / 255.
 	y = torch.from_numpy(np.transpose(y[:, :, [2, 1, 0]], (2, 0, 1))).float()
 	y = y.unsqueeze(0).to(device)