Update app.py

app.py

@@ -1,14 +1,21 @@
 import gradio as gr
-import yolov5
 import os
+import torch
 from transformers import pipeline
 
 imageClassifier = pipeline(task="image-classification",
                            model="PranomVignesh/Police-vs-Public")
 
 
+model = torch.hub.load(
+    'ultralytics/yolov5',
+    'custom',
+    path='./new_best.pt',
+    device="cpu",
+    force_reload=True
+)
+model.eval()
 
-model = yolov5.load('./best.pt', device="cpu")
 
 def predict(image):
     results = model([image], size=224)