drkareemkamal commited on
Commit
d5c47b1
1 Parent(s): cef00bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ def set_custom_prompt():
46
  return prompt
47
 
48
  # Load LLM model
49
- llm_model = CTransformers(model_name='TheBloke/Llama-2-7B-Chat-GGML',
50
  model_type = 'llama',
51
  max_new_token = 512,
52
  temperature=0.5)
 
46
  return prompt
47
 
48
  # Load LLM model
49
+ llm_model = CTransformers(model='TheBloke/Llama-2-7B-Chat-GGML',
50
  model_type = 'llama',
51
  max_new_token = 512,
52
  temperature=0.5)