AventIQ-AI
/

gpt2-next-word-prediction

Model card Files Files and versions Community

ayushsinha commited on Feb 26

Commit

6b4de6f

·

verified ·

1 Parent(s): bb93a87

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -23,7 +23,7 @@ pip install transformers torch
 ### Loading the Model
 ```python
-from sentence_transformers import GPT2LMHeadModel, GPT2Tokenizer
 import torch
 device = "cuda" if torch.cuda.is_available() else "cpu"

 ### Loading the Model
 ```python
+from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import torch
 device = "cuda" if torch.cuda.is_available() else "cpu"