Update app.py
Browse files
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([
|
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 |
|