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:"family sedan", 1:"fire engine",
|
| 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:"bus", 2:"fire engine",3:"heavy truck", 4:"jeep", 5:"mini bus", 6:"racing car", 7:"SUV", 8:"taxi", 9:"truck" }
|
| 15 |
names=[]
|
| 16 |
for result in results:
|
| 17 |
probs = result.probs.top1
|