akhud commited on
Commit
5d7b838
1 Parent(s): f6f3d5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,10 +9,10 @@ def getLLamaresponse(input_text,no_words,blog_style):
9
 
10
  # Load model directly
11
 
12
- model = AutoModel.from_pretrained("TheBloke/Llama-2-7B-Chat-GGML")
13
 
14
  ### LLama2 model
15
- llm=CTransformers(model=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})