Hugging Face's logo

Spaces:
Cidaut
/
Under_haze_detection
Runtime error

App Files Community
paugar commited on
Commit
7586121
·
verified ·
1 Parent(s): 47bab4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ from ultralytics import YOLO
12
 
13
  device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
14
 
15
- model = YOLO('Models/haze_detection.pt', map_location=device)
16
 
17
  model = model.to(device)
18
 
 
12
 
13
  device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
14
 
15
+ model = YOLO('Models/haze_detection.pt')
16
 
17
  model = model.to(device)
18