Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -119,7 +119,7 @@ Task content: {content}
|
|
119 |
'''
|
120 |
|
121 |
# Initialize the LLM
|
122 |
-
llm_model = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.
|
123 |
|
124 |
# Define the prompt template
|
125 |
prompt = PromptTemplate(input_variables=['task_type', 'task_number', 'question', 'content', 'description'], template=initial_prompt)
|
|
|
119 |
'''
|
120 |
|
121 |
# Initialize the LLM
|
122 |
+
llm_model = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.4, top_p=0.85)
|
123 |
|
124 |
# Define the prompt template
|
125 |
prompt = PromptTemplate(input_variables=['task_type', 'task_number', 'question', 'content', 'description'], template=initial_prompt)
|