storresbusquets commited on
Commit
d84d90d
1 Parent(s): 53f76b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,10 +4,11 @@ import transformers
4
  from transformers import AutoTokenizer
5
  from langchain import LLMChain, HuggingFacePipeline, PromptTemplate
6
 
 
7
  def greet(text):
8
 
9
  model = "meta-llama/Llama-2-7b-chat-hf"
10
- tokenizer = AutoTokenizer.from_pretrained(model)
11
 
12
  pipeline = transformers.pipeline(
13
  "text-generation",
 
4
  from transformers import AutoTokenizer
5
  from langchain import LLMChain, HuggingFacePipeline, PromptTemplate
6
 
7
+
8
  def greet(text):
9
 
10
  model = "meta-llama/Llama-2-7b-chat-hf"
11
+ tokenizer = AutoTokenizer.from_pretrained(model, token=llama2)
12
 
13
  pipeline = transformers.pipeline(
14
  "text-generation",