Update app.py

app.py

@@ -22,7 +22,7 @@ from io import BytesIO
 # Load the YOLO model
 from models.common import DetectMultiBackend
 
-weights_path = "./best.pt"
+weights_path = "./best.torchscript"
 device = torch.device("cpu")  # Correctly define the device
 model = torch.jit.load(weights_path)
 # model.eval()  # Load YOLOv5 model correctly