Add fix for output to PIL image.

app.py

@@ -45,8 +45,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))
+    # pred = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
+    # print(type(pred))
     return Image.fromarray(pred)