miguelflor commited on
Commit
2050850
·
verified ·
1 Parent(s): 01b263e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ app.add_middleware(
21
 
22
  # Load your model
23
  model = ResPoseNet()
24
- model.load_state_dict(torch.load('posev0.01126.pth', map_location=torch.device('cpu')))
25
 
26
  def predict_keypoints(image,model):
27
 
 
21
 
22
  # Load your model
23
  model = ResPoseNet()
24
+ model.load_state_dict(torch.load('model.pth', map_location=torch.device('cpu')))
25
 
26
  def predict_keypoints(image,model):
27