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_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]
+>>>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
 
 >>> generate_text("Should I Invest in stocks")
 
-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."
+It's good
 ```