rdlf commited on
Commit
3dbb79e
·
verified ·
1 Parent(s): cc07ecd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ import gradio as gr
3
  # Load model directly
4
  from transformers import AutoTokenizer, AutoModelForCausalLM
5
 
6
- tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
7
- model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
8
 
9
  def eval_text(text):
10
  # Encode the input text
 
3
  # Load model directly
4
  from transformers import AutoTokenizer, AutoModelForCausalLM
5
 
6
+ tokenizer = AutoTokenizer.from_pretrained("NousResearch/Llama-2-7b-chat-hf") #meta-llama/Llama-2-7b-hf
7
+ model = AutoModelForCausalLM.from_pretrained("NousResearch/Llama-2-7b-chat-hf")
8
 
9
  def eval_text(text):
10
  # Encode the input text