Amirizaniani commited on
Commit
6ff6d9e
1 Parent(s): f163cb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -12,8 +12,7 @@ def generate_prompts(user_input):
12
  )
13
  config = {'max_new_tokens': 2048, 'temperature': 0.7, 'context_length': 4096}
14
  llm = CTransformers(model="TheBloke/Mistral-7B-Instruct-v0.1-GGUF",
15
- config=config,
16
- threads=os.cpu_count())
17
  hub_chain = LLMChain(prompt = prompt_template, llm = llm)
18
 
19
  input_data = {"Question": user_input}
 
12
  )
13
  config = {'max_new_tokens': 2048, 'temperature': 0.7, 'context_length': 4096}
14
  llm = CTransformers(model="TheBloke/Mistral-7B-Instruct-v0.1-GGUF",
15
+ config=config)
 
16
  hub_chain = LLMChain(prompt = prompt_template, llm = llm)
17
 
18
  input_data = {"Question": user_input}