vasevooo commited on
Commit
69920f8
1 Parent(s): 47bccad

Update pages/gpt.py

Browse files
Files changed (1) hide show
  1. pages/gpt.py +1 -1
pages/gpt.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.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('models/model.pt', map_location=torch.device('cpu')))
26
 
27
 
28
  length = st.sidebar.slider('**Generated sequence length:**', 8, 256, 15)