Update README.md

README.md

@@ -14,10 +14,10 @@ To load the model:
 (necessary packages: !pip install transformers sentencepiece)
 ```python
 # Load model directly
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-tokenizer = AutoTokenizer.from_pretrained("entity2260/pythia-70m")
-model = AutoModelForCausalLM.from_pretrained("entity2260/pythia-70m")
+tokenizer = AutoTokenizer.from_pretrained("entity2260/algebra_linear_1d")
+model = AutoModelForSeq2SeqLM.from_pretrained("entity2260/algebra_linear_1d")
 ```
 
 You can then use this model to solve algebra 1d equations into numbers.