Spaces:
Running
Running
Update app.py
Browse files
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 #
|
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
|