Update app.py
Browse files
app.py
CHANGED
|
@@ -55,7 +55,7 @@ def process_video(video_path, target, progress=gr.Progress()):
|
|
| 55 |
device = torch.device("cpu")
|
| 56 |
model.to(device)
|
| 57 |
|
| 58 |
-
batch_size =
|
| 59 |
batch_frames = []
|
| 60 |
batch_indices = []
|
| 61 |
|
|
|
|
| 55 |
device = torch.device("cpu")
|
| 56 |
model.to(device)
|
| 57 |
|
| 58 |
+
batch_size = 1
|
| 59 |
batch_frames = []
|
| 60 |
batch_indices = []
|
| 61 |
|