singhankur01 commited on
Commit
f3f08db
·
verified ·
1 Parent(s): ccabfb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-1.5-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.
 
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.