Update app.py

app.py

@@ -6,8 +6,7 @@ from transformers import Owlv2Processor, Owlv2ForObjectDetection
 import numpy as np
 import os
 
-# Check if CUDA is available, otherwise use CPU
-device = 'cuda' if torch.cuda.is_available() else 'cpu'
+device = 'cuda'
 
 processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16")
 model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16").to(device)
@@ -65,7 +64,7 @@ def process_video(video_path, target, progress=gr.Progress()):
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
     out = cv2.VideoWriter(output_path, fourcc, output_fps, (int(cap.get(3)), int(cap.get(4))))
 
-    batch_size = 64
+    batch_size = 128
     frames = []
 
     for frame in progress.tqdm(range(frame_count)):