Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -9,10 +9,10 @@ def getLLamaresponse(input_text,no_words,blog_style):
|
|
9 |
|
10 |
# Load model directly
|
11 |
|
12 |
-
|
13 |
|
14 |
### LLama2 model
|
15 |
-
llm=CTransformers(model=
|
16 |
model_type='llama',
|
17 |
config={'max_new_tokens':256,
|
18 |
'temperature':0.01})
|
|
|
9 |
|
10 |
# Load model directly
|
11 |
|
12 |
+
model_1 = AutoModel.from_pretrained("TheBloke/Llama-2-7B-Chat-GGML")
|
13 |
|
14 |
### LLama2 model
|
15 |
+
llm=CTransformers(model=model_1,
|
16 |
model_type='llama',
|
17 |
config={'max_new_tokens':256,
|
18 |
'temperature':0.01})
|