Hugging Face's logo

Spaces:
Gosula
/
Yolo_V3
Sleeping

App Files Community
3
Gosula commited on
Commit
72e5e98
•
1 Parent(s): e8f7780

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -14,6 +14,7 @@ from model import YoloVersion3
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)
 
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)