Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -51,7 +51,7 @@ async def lifespan(app: FastAPI):
|
|
| 51 |
cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-base")
|
| 52 |
# cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-large")
|
| 53 |
ml_models["reranker_compressor"] = CrossEncoderReranker(model=cross_encoder_model, top_n=5)
|
| 54 |
-
ml_models["llm"] = ChatGoogleGenerativeAI(model="gemini-
|
| 55 |
ml_models["prompt_template"] = PromptTemplate.from_template(
|
| 56 |
"""
|
| 57 |
You are an expert decision maker Assistant in the domain such as insurance, legal compliance, human resources, and contract management.
|
|
|
|
| 51 |
cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-base")
|
| 52 |
# cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-large")
|
| 53 |
ml_models["reranker_compressor"] = CrossEncoderReranker(model=cross_encoder_model, top_n=5)
|
| 54 |
+
ml_models["llm"] = ChatGoogleGenerativeAI(model="gemini-2.0-flash", api_key=GOOGLE_API_KEY)
|
| 55 |
ml_models["prompt_template"] = PromptTemplate.from_template(
|
| 56 |
"""
|
| 57 |
You are an expert decision maker Assistant in the domain such as insurance, legal compliance, human resources, and contract management.
|