Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -11,10 +11,8 @@ access_token = os.getenv("HUGGINGFACE_TOKEN")
|
|
11 |
login(token=access_token)
|
12 |
|
13 |
# from transformers import pipeline
|
14 |
-
|
15 |
-
|
16 |
-
model_name = "TheBloke/Llama-2-7B-Chat-GGML"
|
17 |
-
model = AutoModel.from_pretrained(model_name, use_auth_token=access_token)
|
18 |
|
19 |
## Function to get response form the LLama2 Model
|
20 |
def getLLamaResponse(input_text, no_of_words, blog_style):
|
|
|
11 |
login(token=access_token)
|
12 |
|
13 |
# from transformers import pipeline
|
14 |
+
model_name = "karthiksagarn/llama-2-7b-chat"
|
15 |
+
model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=access_token, from_tf=True)
|
|
|
|
|
16 |
|
17 |
## Function to get response form the LLama2 Model
|
18 |
def getLLamaResponse(input_text, no_of_words, blog_style):
|