singhankur01 commited on
Commit
e85f024
·
verified ·
1 Parent(s): bd3768e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -70,7 +70,7 @@ async def lifespan(app: FastAPI):
70
  print("🚀 Initializing models and prompt template...")
71
 
72
  try:
73
- GOOGLE_API_KEY = os.getenv("gemini_api_key")
74
  print("🔑 gemini_api_key:", "FOUND" if GOOGLE_API_KEY else "NOT FOUND")
75
 
76
  if not GOOGLE_API_KEY:
@@ -98,9 +98,8 @@ async def lifespan(app: FastAPI):
98
  # cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-large")
99
  ml_models["reranker_compressor"] = CrossEncoderReranker(model=cross_encoder_model, top_n=9)
100
  ml_models["llm"] = ChatGoogleGenerativeAI(
101
- # model="gemini-1.5-pro",
102
  # model="gemini-2.0-flash",
103
- model="gemini-2.0-pro",
104
  api_key=GOOGLE_API_KEY,
105
  temperature=0.1,
106
  max_output_tokens=300
 
70
  print("🚀 Initializing models and prompt template...")
71
 
72
  try:
73
+ GOOGLE_API_KEY = os.getenv("gemini_api_key3")
74
  print("🔑 gemini_api_key:", "FOUND" if GOOGLE_API_KEY else "NOT FOUND")
75
 
76
  if not GOOGLE_API_KEY:
 
98
  # cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-large")
99
  ml_models["reranker_compressor"] = CrossEncoderReranker(model=cross_encoder_model, top_n=9)
100
  ml_models["llm"] = ChatGoogleGenerativeAI(
101
+ model="gemini-1.5-pro",
102
  # model="gemini-2.0-flash",
 
103
  api_key=GOOGLE_API_KEY,
104
  temperature=0.1,
105
  max_output_tokens=300