Update app.py

app.py

@@ -20,9 +20,15 @@ import matplotlib.pyplot as plt
 from io import BytesIO
 
 # Load the YOLO model
-model_path = "./last.pt"
-model = torch.jit.load(model_path, map_location=torch.device("cpu"))
+from models.common import DetectMultiBackend
+
+
+weights_path = "./last.pt"
+model = DetectMultiBackend(weights_path, device="cpu")  # Loads the YOLOv5 model correctly
 model.eval()
+# model_path = "./last.pt"
+# model = torch.jit.load(model_path, map_location=torch.device("cpu"))
+# model.eval()
 
 transform = transforms.Compose([
     transforms.Resize((640, 640)),