Sharathhebbar24
/

SSH_355M

Text Generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

Sharathhebbar24 commited on Feb 10

Commit

d531d19

•

1 Parent(s): 9447052

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -34,9 +34,9 @@ prompt.
 ```python
 >>> from transformers import AutoTokenizer, AutoModelForCausalLM
->>> model_name = "Sharathhebbar24/Instruct_GPT"
 >>> model = AutoModelForCausalLM.from_pretrained(model_name)
->>> tokenizer = AutoTokenizer.from_pretrained("gpt2-medium")
 >>> 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)

 ```python
 >>> from transformers import AutoTokenizer, AutoModelForCausalLM
+>>> model_name = "Sharathhebbar24/Instruct_GPT_v1"
 >>> model = AutoModelForCausalLM.from_pretrained(model_name)
+>>> 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)