Update app.py

app.py

@@ -207,7 +207,7 @@ import torchvision.transforms as T
 from ultralytics import YOLO
 
 # Load your model
-model = YOLO()
+model = YOLO("Model_IV.pt")
 # model = torch.load("Model_IV.pt")
 # model.eval()
 checkpoint = torch.load("Model_IV.pt")