Update README.md
Browse files
README.md
CHANGED
|
@@ -45,6 +45,7 @@ sampling_params = SamplingParams(temperature=temperature, max_tokens=max_new_tok
|
|
| 45 |
EHR = ['xxxxx1','xxxxx2']
|
| 46 |
texts = [PROMPT_TEMPLATE.format(query=k) for k in EHR]
|
| 47 |
output = model.generate(texts, sampling_params)
|
|
|
|
| 48 |
```
|
| 49 |
|
| 50 |
|
|
|
|
| 45 |
EHR = ['xxxxx1','xxxxx2']
|
| 46 |
texts = [PROMPT_TEMPLATE.format(query=k) for k in EHR]
|
| 47 |
output = model.generate(texts, sampling_params)
|
| 48 |
+
res = json.loads(output[0].outputs[0].text)
|
| 49 |
```
|
| 50 |
|
| 51 |
|