Ashishkr commited on
Commit
7d25a9f
1 Parent(s): d40e708

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -92,9 +92,10 @@ tokenizer.padding_side = "right"
92
  ```python
93
  # Run text generation pipeline with the merged model
94
  prompt = """
95
- 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?"""
 
96
  pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer,do_sample = True, temperature = 0.9, max_length=200)
97
  response = pipe(f"<s>[INST] {prompt} [/INST]")
98
- print(response)
99
 
100
  ```
 
92
  ```python
93
  # Run text generation pipeline with the merged model
94
  prompt = """
95
+ i have a neck pain since 2 days .
96
+ """
97
  pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer,do_sample = True, temperature = 0.9, max_length=200)
98
  response = pipe(f"<s>[INST] {prompt} [/INST]")
99
+ print(response[0]['generated_text'])
100
 
101
  ```