Commit
·
7b8d56f
1
Parent(s):
a466d2a
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,6 +10,7 @@ model = yolov5.load('./best.pt', device="cpu")
|
|
| 10 |
|
| 11 |
def predict(image):
|
| 12 |
results = model([image], size=224)
|
|
|
|
| 13 |
predictions = imageClassifier(image)
|
| 14 |
classMappings = {
|
| 15 |
'police': "Police / Authorized Personnel",
|
|
|
|
| 10 |
|
| 11 |
def predict(image):
|
| 12 |
results = model([image], size=224)
|
| 13 |
+
print(results)
|
| 14 |
predictions = imageClassifier(image)
|
| 15 |
classMappings = {
|
| 16 |
'police': "Police / Authorized Personnel",
|