Hugging Face's logo

Spaces:
Mehedi2
/
Final_Assignment_Template1
Sleeping

App Files Community
Mehedi2 commited on
Commit
76e2a00
·
verified ·
1 Parent(s): 4fef00e

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -2
agent.py CHANGED
@@ -13,10 +13,11 @@ from tabulate import tabulate
13
  # Load environment variables (make sure GEMINI_API_KEY is set in your Space secrets)
14
  load_dotenv()
15
 
16
- # Initialize the Gemini model using Google AI Studio API key
17
  model = LiteLLMModel(
18
- model_id="google_ai/gemini-2.5-flash-preview-09-2025",
19
  api_key=os.getenv("GEMINI_API_KEY"),
 
 
20
  )
21
 
22
 
 
13
  # Load environment variables (make sure GEMINI_API_KEY is set in your Space secrets)
14
  load_dotenv()
15
 
 
16
  model = LiteLLMModel(
17
+ model_id="gemini-2.5-flash-preview-09-2025",
18
  api_key=os.getenv("GEMINI_API_KEY"),
19
+ provider="google_ai",
20
+
21
  )
22
 
23