Update app.py

app.py

@@ -7,7 +7,9 @@ import pandas as pd
 from transformers import pipeline
 
 # Load the YOLOv5 model
-model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
+# Use a local clone of YOLOv5
+yolo_repo = 'ultralytics/yolov5'
+model = torch.hub.load(yolo_repo, 'yolov5s', source='github')
 
 # Load the translation model
 translator = pipeline("translation_en_to_ar", model="Helsinki-NLP/opus-mt-en-ar")
@@ -73,7 +75,7 @@ def detect_and_draw_video(video_path):
                 overall_counts[label] = overall_counts.get(label, 0) + 1
 
             cv2.rectangle(frame, (int(xmin), int(ymin)), (int(xmax), int(ymax)), (255, 0, 0), 2)
-            cv2.putText(frame, f"{model.names[int(class_id)]}: {conf:.2f}", (int(xmin), int(ymin) - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (255, 255, 255), 2)
+            cv2.putText(frame, f"{model.names[int(class_id)]}: {conf:.2f}", (int(xmin), int(ymin) - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (255, 255, 255), 2)
 
         frames.append(frame)