Spaces:

PranomVignesh
/

Detecting-unauthorized-person-with-firearms

Runtime error

PranomVignesh commited on Jun 22, 2023

Commit

a466d2a

1 Parent(s): 159b7b4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from transformers import pipeline
 imageClassifier = pipeline(task="image-classification",
                            model="PranomVignesh/Police-vs-Public")
-model = yolov5.load('./best.pt', device="cuda")
 def predict(image):
     results = model([image], size=224)

 imageClassifier = pipeline(task="image-classification",
                            model="PranomVignesh/Police-vs-Public")
+model = yolov5.load('./best.pt', device="cpu")
 def predict(image):
     results = model([image], size=224)