singhankur01 commited on
Commit
5eba371
·
verified ·
1 Parent(s): 4c94ccb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ async def lifespan(app: FastAPI):
50
 
51
  # Faster LLM with constrained output
52
  ml_models["llm"] = ChatGoogleGenerativeAI(
53
- model="gemini-1.0-pro",
54
  api_key=GOOGLE_API_KEY,
55
  temperature=0.1,
56
  max_output_tokens=300
 
50
 
51
  # Faster LLM with constrained output
52
  ml_models["llm"] = ChatGoogleGenerativeAI(
53
+ model="gemini-1.5-pro",
54
  api_key=GOOGLE_API_KEY,
55
  temperature=0.1,
56
  max_output_tokens=300