Update README.md

README.md

@@ -33,13 +33,13 @@ prompt.
 >>> model_name = "Sharathhebbar24/Instruct_GPT"
 >>> model = AutoModelForCausalLM.from_pretrained(model_name)
 >>> tokenizer = AutoTokenizer.from_pretrained("gpt2-medium")
->>>def generate_review(text):
->>>   inputs = tokenizer("review: " + text, return_tensors="pt", max_lenght=512, padding='max_length', truncation=True,)
->>>   outputs = mod1.generate(inputs['input_ids'], max_length=128, no_repeat_ngram_size=3, num_beams=6, early_stopping=True)
->>>   summary = tokenizer.decode(outputs[0], skip_special_tokens=True)
->>>   return summary
+>>> 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]
 
 >>> generate_text("Should I Invest in stocks")
 
-It's good
+Should I Invest in stocks? Investing in stocks is a great way to diversify your portfolio.  You can invest in stocks based on the market's performance, or you can invest in stocks based on the company's performance.
 ```