Update app.py

app.py

@@ -207,10 +207,10 @@ import torchvision.transforms as T
 from ultralytics import YOLO
 
 # Load your model
-model = YOLO("Model_IV.pt")
+model = YOLO()
 # model = torch.load("Model_IV.pt")
 # model.eval()
-checkpoint = torch.load("model.pt")
+checkpoint = torch.load("Model_IV.pt")
 model.load_state_dict(checkpoint)  # Load the saved weights
 model.eval()  # Set the model to evaluation mode