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