Update README.md

README.md

@@ -95,7 +95,7 @@ tokenizer.padding_side = "right"
 # Run text generation pipeline with the merged model
 prompt = """
 I had a partial root canal roughly 36 hours ago on my lower left rear molar (wisdom teeth extracted years ago). The doctor drilled out the pulp (noting that it was mostly dead) and put on a temporary filling until the root canal specialist could see me. Now, about 36 hours later, Im in quite a bit of pain again, both on the tooth that was operated on and in a molar on the top of my jaw and in the back jaw muscle. My neck is rather swollen too and is tender to the touch. I have plans to travel tomorrow by plane to see some family and go diving. Is my condition something that requires immediate attention? Should I not be flying or diving?"""
-pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=600)
+pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer,do_sample = True, temperature = 0.9, max_length=200)
 response = pipe(f"<s>[INST] {prompt} [/INST]")
 print(response)