fffiloni commited on
Commit
f1b1bf6
·
1 Parent(s): 8bacd36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -147,7 +147,7 @@ def infer():
147
  #img2_batch = torch.stack([frames[1]])
148
 
149
  img1_batch = torch.stack([input_frame_2])
150
- img2_batch = torch.stack([input_diffused])
151
 
152
  print(f"FRAME AFTER stack: {img1_batch}")
153
 
 
147
  #img2_batch = torch.stack([frames[1]])
148
 
149
  img1_batch = torch.stack([input_frame_2])
150
+ img2_batch = torch.stack([input_frame_1])
151
 
152
  print(f"FRAME AFTER stack: {img1_batch}")
153