Spaces:

fffiloni
/

RAFT

Paused

fffiloni commited on Mar 5, 2023

Commit

d60ba1a

1 Parent(s): 29224cf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -207,7 +207,7 @@ def infer():
     transform = T.ToPILImage()
     # convert the tensor to PIL image using above transform
-    img = transform(img1_batch)
     # display the PIL image
     #img.show()

     transform = T.ToPILImage()
     # convert the tensor to PIL image using above transform
+    img = transform(img1_batch[0])
     # display the PIL image
     #img.show()