Update app.py

app.py

@@ -5,7 +5,7 @@ import shutil
 from ultralytics import YOLO
 
 # Initialize YOLO model
-model = YOLO('./best.pt')
+model = YOLO('./best_train2.pt')
 
 # Class to color mapping
 ClassToColorMapping = {