Sharathhebbar24
/

math_gpt2

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Sharathhebbar24 commited on Jan 24

Commit

57b7106

•

1 Parent(s): 911ef42

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -44,7 +44,7 @@ prompt.
 >>> tokenizer = AutoTokenizer.from_pretrained(model_name)
 >>> def generate_text(prompt):
 >>>  inputs = tokenizer.encode(prompt, return_tensors='pt')
->>>  outputs = mod1.generate(inputs, max_length=64, pad_token_id=tokenizer.eos_token_id)
 >>>  generated = tokenizer.decode(outputs[0], skip_special_tokens=True)
 >>>  return generated[:generated.rfind(".")+1]
 >>> prompt = "What structure is classified as a definite lie algebra?"

 >>> tokenizer = AutoTokenizer.from_pretrained(model_name)
 >>> def generate_text(prompt):
 >>>  inputs = tokenizer.encode(prompt, return_tensors='pt')
+>>>  outputs = model.generate(inputs, max_length=64, pad_token_id=tokenizer.eos_token_id)
 >>>  generated = tokenizer.decode(outputs[0], skip_special_tokens=True)
 >>>  return generated[:generated.rfind(".")+1]
 >>> prompt = "What structure is classified as a definite lie algebra?"