Hugging Face's logo

Spaces:
asigalov61
/
Popular-Hook-Transformer
Sleeping

App Files Community
asigalov61 commited on
Commit
7d7060a
·
verified ·
1 Parent(s): 3eaf1f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -116,7 +116,7 @@ print('Loading model checkpoint...')
116
 
117
  model_path = 'Popular_Hook_Transformer_Small_Trained_Model_10869_steps_0.2308_loss_0.9252_acc.pth'
118
 
119
- model.load_state_dict(torch.load(model_path))
120
 
121
  print('=' * 70)
122
 
 
116
 
117
  model_path = 'Popular_Hook_Transformer_Small_Trained_Model_10869_steps_0.2308_loss_0.9252_acc.pth'
118
 
119
+ model.load_state_dict(torch.load(model_path, map_location=DEVICE))
120
 
121
  print('=' * 70)
122