Hugging Face's logo

Spaces:
vvv-knyazeva
/
NLP_project
Runtime error

App Files Community
vvv-knyazeva commited on
Commit
52d02b5
·
1 Parent(s): 2f4f3d6

Update pages/gpt_v1.py

Browse files
Files changed (1) hide show
  1. pages/gpt_v1.py +1 -1
pages/gpt_v1.py CHANGED
@@ -14,7 +14,7 @@ model = GPT2LMHeadModel.from_pretrained(
14
  output_hidden_states = False,
15
  )
16
 
17
- model.load_state_dict(torch.load('models/model.pt', map_location=torch.device('cpu')))
18
 
19
 
20
  length = st.sidebar.slider('**Generated sequence length:**', 8, 256, 15)
 
14
  output_hidden_states = False,
15
  )
16
 
17
+ model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
18
 
19
 
20
  length = st.sidebar.slider('**Generated sequence length:**', 8, 256, 15)