Update app.py

app.py

@@ -80,8 +80,10 @@ def rem_bg(vid):
         yield fbox2,frame_num
 
 
-def im_2_vid():
-    width, height = Image.open(images[0]).size
+def im_2_vid(images,fps):
+    #width, height = Image.open(images[0]).size
+    width, height = cv2.imread(images[0]).shape()
+    print (width,height)
     size = (width, height)
     codec = cv2.VideoWriter_fourcc(*'mp4v') #DIVX, DIVD
     if (len(sys.argv) < 2 and n > 0):