Update app.py

app.py

@@ -23,9 +23,12 @@ from io import BytesIO
 from models.common import DetectMultiBackend
 
 
+
 weights_path = "./last.pt"
-model = DetectMultiBackend(weights_path, device="cpu")  # Loads the YOLOv5 model correctly
+device = torch.device("cpu")  # Correctly define the device
+model = DetectMultiBackend(weights_path, device=device)  # Load YOLOv5 model correctly
 model.eval()
+
 # model_path = "./last.pt"
 # model = torch.jit.load(model_path, map_location=torch.device("cpu"))
 # model.eval()