Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = "
|
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
|