Update app.py

app.py

@@ -34,7 +34,7 @@ def process_video(video_path, target, progress=gr.Progress()):
 
     processed_frames = []
     frame_scores = []
-    batch_size = 32
+    batch_size = 8
     batch_frames = []
     batch_times = []
 
@@ -70,7 +70,7 @@ def process_video(video_path, target, progress=gr.Progress()):
                 boxes, scores, labels = results[i]["boxes"], results[i]["scores"], results[i]["labels"]
 
                 for box, score, label in zip(boxes, scores, labels):
-                    if score.item() >= 0.25:
+                    if score.item() >= 0.5:
                         box = [round(i, 2) for i in box.tolist()]
                         object_label = target
                         confidence = round(score.item(), 3)
@@ -120,7 +120,7 @@ def load_sample_frame(video_path):
 
 def gradio_app():
     with gr.Blocks() as app:
-        gr.Markdown("# Video Object Detection with Owlv2 (3 FPS, Batch Size 32)")
+        gr.Markdown("# Video Object Detection with Owlv2")
 
         video_input = gr.Video(label="Upload Video")
         target_input = gr.Textbox(label="Target Object", value="Elephant")