Hugging Face's logo

Spaces:
Cidaut
/
Under_haze_detection
Runtime error

App Files Community
paugar commited on
Commit
6ead422
·
verified ·
1 Parent(s): ad1c17e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ 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
 
21
  model = model.to(device)
22
 
 
16
 
17
  checkpoints = torch.load('Models/haze_detection.pt', map_location=device)
18
 
19
+ model.load(checkpoints)
20
 
21
  model = model.to(device)
22