Update app.py

app.py

@@ -12,9 +12,7 @@ from ultralytics import YOLO
 
 device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
 
-model = YOLO('Models/yolov8m.pt')
-
-checkpoints = torch.load('Models/haze_detection.pt', map_location=device)
+model =  torch.load('Models/haze_detection.pt', map_location=device)
 
 model.load(checkpoints)