Add fix for output to PIL image.

app.py

@@ -1,5 +1,6 @@
 import cv2
 import torch
+import numpy as np
 import gradio as gr
 from PIL import Image
 from super_image import ImageLoader, EdsrModel, MsrnModel, MdsrModel, AwsrnModel, A2nModel, CarnModel, PanModel, \
@@ -45,9 +46,8 @@ def inference(img, scale_str, model_name):
     preds = model(inputs)
     preds = preds.data.cpu().numpy()
     pred = preds[0].transpose((1, 2, 0)) * 255.0
-    # pred = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
-    # print(type(pred))
-    return Image.fromarray(pred)
+    pred = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
+    return Image.fromarray((pred * 255).astype(np.uint8))
 
 
 torch.hub.download_url_to_file('http://people.rennes.inria.fr/Aline.Roumy/results/images_SR_BMVC12/input_groundtruth/baby_mini_d3_gaussian.bmp',