Update app.py

app.py

@@ -11,7 +11,8 @@ model = YOLO("Vehicles_Classify_v1.pt")
 
 def detect_objects(images):
     results = model(images)
-    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" }
+    # 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" }
+    classes={ 0:"SUV", 1:"bus", 2:"family sedan", 3:"fire engine",4:"heavy truck", 5:"jeep", 6:"mini bus", 7:"racing car", 8:"taxi", 9:"truck" }
     names=[]  
     for result in results:          
         probs = result.probs.top1