Amirizaniani commited on
Commit
09a84bf
·
1 Parent(s): e12f506

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def answer_question(prompt):
34
  template=f"Answer '{prompt} 'and do not consider the number behind it."
35
  )
36
  config = {'max_new_tokens': 512, 'temperature': 0.7, 'context_length': 512}
37
- llm = CTransformers(model="TheBloke/Llama-2-7B-Chat-GGML",
38
  config=config)
39
  hub_chain = LLMChain(prompt = prompt_template, llm = llm)
40
  input_data = {"Question": prompt}
 
34
  template=f"Answer '{prompt} 'and do not consider the number behind it."
35
  )
36
  config = {'max_new_tokens': 512, 'temperature': 0.7, 'context_length': 512}
37
+ llm = CTransformers(model="TheBloke/zephyr-7B-beta-GGUF",
38
  config=config)
39
  hub_chain = LLMChain(prompt = prompt_template, llm = llm)
40
  input_data = {"Question": prompt}