Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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('
|
| 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 |
|