Update app.py

app.py

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