Spaces:

vvv-knyazeva
/

NLP_project

Runtime error

vvv-knyazeva commited on Jun 8, 2023

Commit

3c332cd

1 Parent(s): 9413947

Update pages/gpt_v1.py

Files changed (1) hide show

pages/gpt_v1.py CHANGED Viewed

@@ -13,7 +13,7 @@ model = GPT2LMHeadModel.from_pretrained(
     output_hidden_states = False,
 )
 # Вешаем сохраненные веса на нашу модель
-model.load_state_dict(torch.load('modelgpt.pt', map_location=torch.device('cpu')))
 length = st.sidebar.slider('Длина генерируемой последовательности:', 8, 256, 15)

     output_hidden_states = False,
 )
 # Вешаем сохраненные веса на нашу модель
+model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
 length = st.sidebar.slider('Длина генерируемой последовательности:', 8, 256, 15)