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

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/zephyr-7B-beta-GGUF",
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="mistralai/Mistral-7B-v0.1",
38
  config=config)
39
  hub_chain = LLMChain(prompt = prompt_template, llm = llm)
40
  input_data = {"Question": prompt}