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 |
|