singhankur01 commited on
Commit
3e37c9a
·
verified ·
1 Parent(s): efe185d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -97,8 +97,8 @@ async def lifespan(app: FastAPI):
97
  # cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-large")
98
  ml_models["reranker_compressor"] = CrossEncoderReranker(model=cross_encoder_model, top_n=9)
99
  ml_models["llm"] = ChatGoogleGenerativeAI(
100
- model="gemini-1.5-pro",
101
- # model="gemini-2.0-flash",
102
  api_key=GOOGLE_API_KEY,
103
  temperature=0.1,
104
  max_output_tokens=300
 
97
  # cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-large")
98
  ml_models["reranker_compressor"] = CrossEncoderReranker(model=cross_encoder_model, top_n=9)
99
  ml_models["llm"] = ChatGoogleGenerativeAI(
100
+ # model="gemini-1.5-pro",
101
+ model="gemini-2.0-flash",
102
  api_key=GOOGLE_API_KEY,
103
  temperature=0.1,
104
  max_output_tokens=300