Alexvatti commited on
Commit
a00f794
Β·
verified Β·
1 Parent(s): a68cbfc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,7 +9,7 @@ from huggingface_hub import login
9
  login(token=os.environ["HUGGINGFACE_TOKEN"])
10
 
11
  # Model and Tokenizer
12
- model_name = "meta-llama/Meta-Llama-3-8B" # Change to 13B or 70B if needed
13
 
14
  tokenizer = AutoTokenizer.from_pretrained(model_name,use_auth_token=True)
15
 
@@ -37,5 +37,5 @@ def generate_text(prompt):
37
  def chat_with_llama(prompt):
38
  return generate_text(prompt)
39
 
40
- gr.Interface(fn=chat_with_llama, inputs="text", outputs="text", title="LLaMA 3 Chatbot").launch()
41
 
 
9
  login(token=os.environ["HUGGINGFACE_TOKEN"])
10
 
11
  # Model and Tokenizer
12
+ model_name = "meta-llama/Llama-2-7b" # Change to 13B or 70B if needed
13
 
14
  tokenizer = AutoTokenizer.from_pretrained(model_name,use_auth_token=True)
15
 
 
37
  def chat_with_llama(prompt):
38
  return generate_text(prompt)
39
 
40
+ gr.Interface(fn=chat_with_llama, inputs="text", outputs="text", title="LLaMA 2 Chatbot").launch()
41