Spaces:
Sleeping
Sleeping
Update TextGen/router.py
Browse files- TextGen/router.py +1 -1
TextGen/router.py
CHANGED
@@ -84,7 +84,7 @@ def generate_text(messages: List[str], npc:str):
|
|
84 |
# Initialize the LLM
|
85 |
llm = ChatGoogleGenerativeAI(
|
86 |
model="gemini-1.5-pro-latest",
|
87 |
-
max_output_tokens=
|
88 |
temperature=1,
|
89 |
safety_settings={
|
90 |
HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
|
|
|
84 |
# Initialize the LLM
|
85 |
llm = ChatGoogleGenerativeAI(
|
86 |
model="gemini-1.5-pro-latest",
|
87 |
+
max_output_tokens=100,
|
88 |
temperature=1,
|
89 |
safety_settings={
|
90 |
HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
|