Update README.md
Browse files
README.md
CHANGED
@@ -94,32 +94,6 @@ Respond as succinctly as possible. Format the response as a completion of this t
|
|
94 |
|:---|:----------|:--------|:-----:|"""
|
95 |
|
96 |
|
97 |
-
messages = [
|
98 |
-
{"role": "system", "content": "You are a helpful assistant."},
|
99 |
-
{"role": "user", "content": question},
|
100 |
-
]
|
101 |
-
prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
102 |
-
# print("***Prompt:\n", prompt)
|
103 |
-
|
104 |
-
outputs = pipeline(prompt, max_new_tokens=1000, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
|
105 |
-
print("***Generation:")
|
106 |
-
print(outputs[0]["generated_text"][len(prompt):])
|
107 |
-
|
108 |
-
```
|
109 |
-
|
110 |
-
```
|
111 |
-
***Generation:
|
112 |
-
|1|Calculate total sold|Add morning and afternoon sales|132|
|
113 |
-
|2|Subtract sold from total|200 - 132|68|
|
114 |
-
|3|Adjust for returns|Add returned loaves to remaining|74|
|
115 |
-
```
|
116 |
-
|
117 |
-
### Example 1
|
118 |
-
|
119 |
-
```python
|
120 |
-
question = """How many r's are in the word "strawberry"?""
|
121 |
-
|
122 |
-
|
123 |
messages = [
|
124 |
{"role": "system", "content": "You are a helpful assistant."},
|
125 |
{"role": "user", "content": question},
|
|
|
94 |
|:---|:----------|:--------|:-----:|"""
|
95 |
|
96 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
97 |
messages = [
|
98 |
{"role": "system", "content": "You are a helpful assistant."},
|
99 |
{"role": "user", "content": question},
|