Update app.py

app.py

@@ -24,7 +24,8 @@ from models.common import DetectMultiBackend
 
 weights_path = "./last.pt"
 device = torch.device("cpu")  # Correctly define the device
-model = DetectMultiBackend(weights_path, device=device)  # Load YOLOv5 model correctly
+model = torch.jit.load(weights_path)
+# model.eval()  # Load YOLOv5 model correctly
 model.eval()
 
 # model_path = "./last.pt"