Add fix for output to PIL image.

app.py

@@ -1,7 +1,7 @@
 import cv2
 import torch
 import gradio as gr
-from torchvision import transforms
+from PIL import Image
 from super_image import ImageLoader, EdsrModel, MsrnModel, MdsrModel, AwsrnModel, A2nModel, CarnModel, PanModel, \
     HanModel, DrlnModel, RcanModel
 
@@ -46,7 +46,7 @@ def inference(img, scale_str, model_name):
     preds = preds.data.cpu().numpy()
     pred = preds[0].transpose((1, 2, 0)) * 255.0
     pred = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
-    return transforms.ToPILImage(mode='RGB')(pred)
+    return Image.fromarray(pred)
 
 
 torch.hub.download_url_to_file('http://people.rennes.inria.fr/Aline.Roumy/results/images_SR_BMVC12/input_groundtruth/baby_mini_d3_gaussian.bmp',