reab5555 commited on
Commit
d6151e4
·
verified ·
1 Parent(s): c9f1714

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = 4 # Process 4 frames at a time
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