eddyejembi commited on
Commit
4d1a0d0
·
1 Parent(s): dd86232

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -4,9 +4,11 @@ import torch
4
  from transformers import pipeline
5
  import gradio as gr
6
 
7
- model = "meta-llama/Llama-2-7b-chat-hf"
8
  access_token = "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"
9
- tokenizer = AutoTokenizer.from_pretrained(model, token=access_token)
 
 
10
 
11
  pipeline = transformers.pipeline(
12
  "text-generation",
 
4
  from transformers import pipeline
5
  import gradio as gr
6
 
7
+ llama_model = "meta-llama/Llama-2-7b-chat-hf"
8
  access_token = "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"
9
+
10
+ model = AutoModel.from_pretrained(llama_model, token=access_token)
11
+ tokenizer = AutoTokenizer.from_pretrained(llama_model, token=access_token)
12
 
13
  pipeline = transformers.pipeline(
14
  "text-generation",