Spaces:

fffiloni
/

RAFT

Paused

fffiloni commited on Mar 8, 2023

Commit

a99d573

1 Parent(s): f6486fd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -148,7 +148,7 @@ def infer():
     #img2_batch = torch.stack([frames[1]])
     img1_batch = torch.stack([input_frame_2])
-    img2_batch = torch.stack([input_frame_1])
     print(f"FRAME AFTER stack: {img1_batch}")

     #img2_batch = torch.stack([frames[1]])
     img1_batch = torch.stack([input_frame_2])
+    img2_batch = torch.stack([input_diffused])
     print(f"FRAME AFTER stack: {img1_batch}")