ryanrwatkins commited on
Commit
4eddc47
1 Parent(s): f0a6f2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -133,8 +133,8 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
133
  gen_ai = ChatGoogleGenerativeAI(model="gemini-pro")
134
  response = gen_ai.invoke(
135
  input=query, # Pass both history and current prompt
136
- max_tokens=max_tokens,
137
- temperature=temperature # Adjust temperature as needed
138
  )
139
 
140
  completion = response.content # Extract the completion message
 
133
  gen_ai = ChatGoogleGenerativeAI(model="gemini-pro")
134
  response = gen_ai.invoke(
135
  input=query, # Pass both history and current prompt
136
+ #max_tokens=max_tokens, # for open ai only
137
+ #temperature=temperature # for open ai only
138
  )
139
 
140
  completion = response.content # Extract the completion message