remove output of original image

app.py

@@ -95,7 +95,7 @@ def pred_SRCNN(model, image, device, scale_factor=2):
         (original_size[1] * scale_factor, original_size[0] * scale_factor),
         interpolation=transforms.InterpolationMode.BICUBIC,
     )(image)
-    return out_final, image_bicubic, image
+    return out_final, image_bicubic
 
 
 # load model
@@ -132,7 +132,7 @@ def super_reso(image):
 
     # prediction
     with torch.no_grad():
-        out_final, image_bicubic, image = pred_SRCNN(
+        out_final, image_bicubic = pred_SRCNN(
             model=model, image=image, device=device
         )
     # grid = image_grid([out_final,image_bicubic],1,2)