Update README.md

README.md

@@ -50,8 +50,8 @@ print(tokenizer.decode(outputs[0]))
 # pip install accelerate
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 
-tokenizer = T5Tokenizer.from_pretrained("ArtifactAI/flan-t5-base-arxiv-cs-ml-question-answering")
-model = T5ForConditionalGeneration.from_pretrained("ArtifactAI/flan-t5-base-arxiv-cs-ml-question-answering", device_map="auto")
+tokenizer = T5Tokenizer.from_pretrained("ArtifactAI/flan-t5-base-arxiv-math-question-answering")
+model = T5ForConditionalGeneration.from_pretrained("ArtifactAI/flan-t5-base-arxiv-math-question-answering", device_map="auto")
 
 input_text = "What is the spectral isolation of bi-invariant metrics?"
 input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")
@@ -67,7 +67,7 @@ print(tokenizer.decode(outputs[0]))
 
 ```python
 # load model and tokenizer from huggingface hub with pipeline
-qa = pipeline("summarization", model="ArtifactAI/flan-t5-base-arxiv-cs-ml-question-answering")
+qa = pipeline("summarization", model="ArtifactAI/flan-t5-base-arxiv-math-question-answering")
 
 
 query = "What is the spectral isolation of bi-invariant metrics?"