Add fix for output to PIL image.

app.py

@@ -46,7 +46,6 @@ 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)
     return Image.fromarray(pred.astype('uint8'), 'RGB')
 
 
@@ -56,6 +55,7 @@ torch.hub.download_url_to_file('http://people.rennes.inria.fr/Aline.Roumy/result
                                'woman.bmp')
 torch.hub.download_url_to_file('http://people.rennes.inria.fr/Aline.Roumy/results/images_SR_BMVC12/input_groundtruth/bird_mini_d4_gaussian.bmp',
                                'bird.bmp')
+
 models = ['EDSR-base', 'DRLN', 'EDSR', 'MSRN', 'MDSR', 'AWSRN-BAM', 'A2N', 'CARN', 'PAN']
 scales = [2, 3, 4]
 for model_name in models: