Update app.py

app.py

@@ -11,7 +11,8 @@ model = YOLO("Bottles_Cans_Classify_v1.pt")
 
 def detect_objects(images):
     results = model(images)
-    classes={ 0:"bottle_25cl", 1:"bottle_33cl", 2:"bottle_50cl", 3:"bottle_100cl", 4:"bottle_150cl", 5:"bottle_200cl", 6:"can", 7:"reject" }
+    classes={ 0:"Reject", 1:"bottle_100cl", 2:"bottle_150cl", 3:"bottle_200cl", 4:"bottle_25cl", 5:"bottle_33cl", 6:"bottle_50cl", 7:"can"  }
+    # classes={ 0:"bottle_25cl", 1:"bottle_33cl", 2:"bottle_50cl", 3:"bottle_100cl", 4:"bottle_150cl", 5:"bottle_200cl", 6:"can", 7:"reject" }
     names=[]  
     for result in results:          
         probs = result.probs.top1