Hugging Face's logo

Spaces:
jeydipak
/
nlp_project
Runtime error

App Files Community
2
jeydipak commited on
Commit
db07551
·
verified ·
1 Parent(s): b0c0dbc

Update pages/task3.py

Browse files
Files changed (1) hide show
  1. pages/task3.py +1 -1
pages/task3.py CHANGED
@@ -33,7 +33,7 @@ DEVICE = 'cpu'
33
 
34
  tokenizer_path = "sberbank-ai/rugpt3small_based_on_gpt2"
35
 
36
- model = torch.load('srcs/gpt_weights.pth', map_location=torch.device('cpu')).to(DEVICE)
37
  tokenizer = GPT2Tokenizer.from_pretrained(tokenizer_path)
38
 
39
  st.markdown("""
 
33
 
34
  tokenizer_path = "sberbank-ai/rugpt3small_based_on_gpt2"
35
 
36
+ model = torch.load('srcs/gpt_weights.pth', map_location=torch.device('cpu'))
37
  tokenizer = GPT2Tokenizer.from_pretrained(tokenizer_path)
38
 
39
  st.markdown("""