Spaces:

fffiloni
/

RAFT

Paused

fffiloni commited on Mar 28, 2023

Commit

0e27115

1 Parent(s): f252d28

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -84,8 +84,8 @@ def infer():
     #img1_batch = torch.stack([frames[0]])
     #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}")

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