storresbusquets commited on
Commit
8854b4b
·
1 Parent(s): 761feb6

Update app.py

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