Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ model = YOLO("Vehicles_Classify_v1.pt")
|
|
| 11 |
|
| 12 |
def detect_objects(images):
|
| 13 |
results = model(images)
|
| 14 |
-
classes={ 0:"
|
| 15 |
names=[]
|
| 16 |
for result in results:
|
| 17 |
probs = result.probs.top1
|
|
|
|
| 11 |
|
| 12 |
def detect_objects(images):
|
| 13 |
results = model(images)
|
| 14 |
+
classes={ 0:"family sedan", 1:"fire engine", 2:"heavy truck", 3:"jeep", 4:"mini bus", 5:"racing car", 6:"SUV", 7:"bus", 8:"taxi", 9:"truck" }
|
| 15 |
names=[]
|
| 16 |
for result in results:
|
| 17 |
probs = result.probs.top1
|