charlesdedampierre commited on
Commit
8fffa2b
·
verified ·
1 Parent(s): 3202359

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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 a Large Language Model?"}
74
  ]
75
- tokenizer = AutoTokenizer.from_pretrained(new_model)
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=new_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