Sharathhebbar24
/

math_gpt2_sft

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Sharathhebbar24 commited on Jan 24

Commit

3ed7a2e

•

1 Parent(s): 5c2496e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -38,7 +38,7 @@ prompt.
 ```python
 >>> from transformers import AutoTokenizer, AutoModelForCausalLM
->>> model_name = "Sharathhebbar24/math_gpt2"
 >>> model = AutoModelForCausalLM.from_pretrained(model_name)
 >>> tokenizer = AutoTokenizer.from_pretrained(model_name)
 >>> def generate_text(prompt):

 ```python
 >>> from transformers import AutoTokenizer, AutoModelForCausalLM
+>>> model_name = "Sharathhebbar24/math_gpt2_sft"
 >>> model = AutoModelForCausalLM.from_pretrained(model_name)
 >>> tokenizer = AutoTokenizer.from_pretrained(model_name)
 >>> def generate_text(prompt):