Update app.py

app.py

@@ -10,13 +10,12 @@ import matplotlib.pyplot as plt
 import tempfile
 import shutil
 
-# Check if CUDA is available, otherwise use CPU
-device = "cpu"
+device = "cuda"
 
 processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16")
 model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16")
 
-model = model.to(device).half()
+model = model.to(device)
 
 def process_video(video_path, target, progress=gr.Progress()):
     if video_path is None: