Tirath5504 commited on
Commit
78d1113
·
verified ·
1 Parent(s): 0e3f84a

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -43,7 +43,7 @@ def stream_object_detection(video, conf_threshold):
43
  if n_frames % SUBSAMPLE == 0:
44
  batch.append(frame)
45
  if len(batch) == 2 * desired_fps:
46
- inputs = image_processor(images=batch, return_tensors="pt").to("cuda")
47
 
48
  print(f"starting batch of size {len(batch)}")
49
  start = time.time()
 
43
  if n_frames % SUBSAMPLE == 0:
44
  batch.append(frame)
45
  if len(batch) == 2 * desired_fps:
46
+ inputs = image_processor(images=batch, return_tensors="pt")#.to("cuda")
47
 
48
  print(f"starting batch of size {len(batch)}")
49
  start = time.time()