ArielDrabkin commited on
Commit
3875486
·
verified ·
1 Parent(s): 2565492

Update SolarPanelDetector.py

Browse files
Files changed (1) hide show
  1. SolarPanelDetector.py +0 -2
SolarPanelDetector.py CHANGED
@@ -2,9 +2,7 @@ from ultralytics import YOLO
2
  from PIL import Image
3
  import requests
4
  import random
5
- import torch
6
  import ultralytics
7
- torch.serialization.add_safe_globals(ultralytics.nn.tasks.DetectionModel)
8
 
9
  model = YOLO('detector.pt')
10
 
 
2
  from PIL import Image
3
  import requests
4
  import random
 
5
  import ultralytics
 
6
 
7
  model = YOLO('detector.pt')
8