Spaces:

fffiloni
/

RAFT

Paused

fffiloni commited on Mar 5, 2023

Commit

c34fc9f

1 Parent(s): c798d8a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -170,7 +170,7 @@ def infer():
     transform = T.ToPILImage()
     # convert the tensor to PIL image using above transform
-    img = transform(input_frame_2)
     img = img.resize((960, 520))
     # display the PIL image
     #img.show()

     transform = T.ToPILImage()
     # convert the tensor to PIL image using above transform
+    img = transform(frames[1])
     img = img.resize((960, 520))
     # display the PIL image
     #img.show()