Update README.md
Browse files
README.md
CHANGED
@@ -70,15 +70,15 @@ from transformers import AutoTokenizer
|
|
70 |
# Format prompt
|
71 |
message = [
|
72 |
{"role": "system", "content": "You are a helpful assistant chatbot."},
|
73 |
-
{"role": "user", "content": "What is
|
74 |
]
|
75 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
76 |
prompt = tokenizer.apply_chat_template(message, add_generation_prompt=True, tokenize=False)
|
77 |
|
78 |
# Create pipeline
|
79 |
pipeline = transformers.pipeline(
|
80 |
"text-generation",
|
81 |
-
model=
|
82 |
tokenizer=tokenizer
|
83 |
)
|
84 |
|
|
|
70 |
# Format prompt
|
71 |
message = [
|
72 |
{"role": "system", "content": "You are a helpful assistant chatbot."},
|
73 |
+
{"role": "user", "content": "What is Topic Modeling?"}
|
74 |
]
|
75 |
+
tokenizer = AutoTokenizer.from_pretrained('charlesdedampierre/TopicNeuralHermes-2.5-Mistral-7B')
|
76 |
prompt = tokenizer.apply_chat_template(message, add_generation_prompt=True, tokenize=False)
|
77 |
|
78 |
# Create pipeline
|
79 |
pipeline = transformers.pipeline(
|
80 |
"text-generation",
|
81 |
+
model='charlesdedampierre/TopicNeuralHermes-2.5-Mistral-7B',
|
82 |
tokenizer=tokenizer
|
83 |
)
|
84 |
|