Hugging Face's logo

Spaces:
Gosula
/
Yolo_V3
Sleeping

App Files Community
3
Gosula commited on
Commit
82b337d
•
1 Parent(s): 2a52a09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -14,7 +14,6 @@ from model import YoloVersion3
14
  import matplotlib.pyplot as plt
15
  import matplotlib.patches as patches
16
 
17
-
18
  # Load the model
19
  model = YoloVersion3( )
20
  model.load_state_dict(torch.load('Yolov3.pth', map_location=torch.device('cpu')), strict=False)
 
14
  import matplotlib.pyplot as plt
15
  import matplotlib.patches as patches
16
 
 
17
  # Load the model
18
  model = YoloVersion3( )
19
  model.load_state_dict(torch.load('Yolov3.pth', map_location=torch.device('cpu')), strict=False)