Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -75,7 +75,7 @@ async def lifespan(app: FastAPI):
|
|
| 75 |
ml_models["reranker_compressor"] = CrossEncoderReranker(model=cross_encoder_model, top_n=6)
|
| 76 |
ml_models["llm"] = ChatGoogleGenerativeAI(
|
| 77 |
# model="gemini-1.5-pro",
|
| 78 |
-
model="gemini-2.0-flash,
|
| 79 |
api_key=GOOGLE_API_KEY,
|
| 80 |
temperature=0.1,
|
| 81 |
max_output_tokens=300
|
|
|
|
| 75 |
ml_models["reranker_compressor"] = CrossEncoderReranker(model=cross_encoder_model, top_n=6)
|
| 76 |
ml_models["llm"] = ChatGoogleGenerativeAI(
|
| 77 |
# model="gemini-1.5-pro",
|
| 78 |
+
model="gemini-2.0-flash",
|
| 79 |
api_key=GOOGLE_API_KEY,
|
| 80 |
temperature=0.1,
|
| 81 |
max_output_tokens=300
|