Hugging Face's logo

Spaces:
vvv-knyazeva
/
NLP_project
Runtime error

App Files Community
vvv-knyazeva commited on
Commit
9c4d73b
•
1 Parent(s): 765d0d4

Update pages/gpt_v1.py

Browse files
Files changed (1) hide show
  1. pages/gpt_v1.py +1 -1
pages/gpt_v1.py CHANGED
@@ -22,7 +22,7 @@ model = GPT2LMHeadModel.from_pretrained(
22
  output_hidden_states = False,
23
  )
24
 
25
- model.load_state_dict(torch.load('model(1).pt', map_location=torch.device('cpu')))
26
 
27
 
28
  length = st.sidebar.slider('**Generated sequence length:**', 8, 256, 15)
 
22
  output_hidden_states = False,
23
  )
24
 
25
+ model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
26
 
27
 
28
  length = st.sidebar.slider('**Generated sequence length:**', 8, 256, 15)