Chandranshu Jain commited on
Commit
ef57dfe
1 Parent(s): 261e2dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -71,7 +71,8 @@ def get_conversational_chain():
71
 
72
  Answer:
73
  """
74
- model = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.3, google_api_key=GOOGLE_API_KEY)
 
75
  prompt = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
76
  chain = load_qa_chain(model, chain_type="stuff", prompt=prompt)
77
  return chain
 
71
 
72
  Answer:
73
  """
74
+ #model = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.3, google_api_key=GOOGLE_API_KEY)
75
+ model = ChatGoogleGenerativeAI(model="gemini-1.0-pro-latest", temperature=0.3, google_api_key=GOOGLE_API_KEY)
76
  prompt = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
77
  chain = load_qa_chain(model, chain_type="stuff", prompt=prompt)
78
  return chain