Amirizaniani commited on
Commit
fe6c5d7
·
1 Parent(s): f954458

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def generate_prompts(user_input):
15
  template=f"Just list 10 quetion prompts for {user_input} and don't put number before each of the prompts."
16
  )
17
  config = {'max_new_tokens': 512, 'temperature': 0.7, 'context_length': 512}
18
- llm = CTransformers(model="HuggingFaceH4/zephyr-7b-alpha",
19
  config=config)
20
  hub_chain = LLMChain(prompt = prompt_template, llm = llm)
21
  input_data = {"Question": user_input}
 
15
  template=f"Just list 10 quetion prompts for {user_input} and don't put number before each of the prompts."
16
  )
17
  config = {'max_new_tokens': 512, 'temperature': 0.7, 'context_length': 512}
18
+ llm = CTransformers(model="TheBloke/zephyr-7B-alpha-GGUF",
19
  config=config)
20
  hub_chain = LLMChain(prompt = prompt_template, llm = llm)
21
  input_data = {"Question": user_input}