stelladai commited on
Commit
b62807b
1 Parent(s): 45d5016

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import torch
7
  hugging_face_token = os.getenv('HUGGINGFACEHUB_API_TOKEN')
8
 
9
  def getLLamaresponse(input_text, no_words, blog_style):
10
- model = "meta-llama/Llama-2-7b-chat-hf"
11
  tokenizer = AutoTokenizer.from_pretrained(model, use_auth_token=hugging_face_token)
12
 
13
  # Configure the model's device (CPU or GPU) based on the environment
 
7
  hugging_face_token = os.getenv('HUGGINGFACEHUB_API_TOKEN')
8
 
9
  def getLLamaresponse(input_text, no_words, blog_style):
10
+ model = "daryl149/llama-2-7b-chat-hf"
11
  tokenizer = AutoTokenizer.from_pretrained(model, use_auth_token=hugging_face_token)
12
 
13
  # Configure the model's device (CPU or GPU) based on the environment