Update app.py

app.py

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