Hugging Face's logo

Spaces:
Cidaut
/
Under_haze_detection
Runtime error

App Files Community
paugar commited on
Commit
30b597c
·
verified ·
1 Parent(s): e4b9b22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cp
14
 
15
  model = YOLO('Models/yolov8m.pt')
16
 
17
- checkpoints = torch.load('Models/dehaze_yolo.pt', map_location=device)
18
 
19
  model.load_state_dict(checkpoints['model_state_dict'])
20
 
 
14
 
15
  model = YOLO('Models/yolov8m.pt')
16
 
17
+ checkpoints = torch.load('Models/haze_detection.pt', map_location=device)
18
 
19
  model.load_state_dict(checkpoints['model_state_dict'])
20