Mr-Vicky-01 commited on
Commit
8357cfe
1 Parent(s): e5d3bd2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def get_conversational_chain():
49
  model = HuggingFaceHub(repo_id="google/gemma-1.1-7b-it",
50
  model_kwargs={"temperature": 0.2,"max_new_tokens":512, "return_only_answer":True})
51
  prompt = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
52
- chain = LLMChain(llm=model, chain_type="stuff", prompt=prompt)
53
  return chain
54
 
55
  def user_input(user_question):
 
49
  model = HuggingFaceHub(repo_id="google/gemma-1.1-7b-it",
50
  model_kwargs={"temperature": 0.2,"max_new_tokens":512, "return_only_answer":True})
51
  prompt = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
52
+ chain = LLMChain(llm=model, prompt=prompt)
53
  return chain
54
 
55
  def user_input(user_question):