Update app.py

app.py

@@ -22,6 +22,7 @@ import numpy as np
 import os
 import sys
 import torch
+from PIL import Image
 import matplotlib.pyplot as plt
 import torchvision.transforms.functional as F
 from torchvision.io import read_video
@@ -227,6 +228,7 @@ def infer():
     # res = cv2.remap(img, flow, None, cv2.INTER_LINEAR)
     res = cv2.remap(frame1pil, numpy_array_flow, None, cv2.INTER_LANCZOS4)
     print(res)
+    res = Image.fromarray(res)
     res.save('wraped.jpg')
     return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg', 'wraped.jpg'
 ####################################