ChrisSacrumCor commited on
Commit
d41b8a0
·
verified ·
1 Parent(s): 2496ce1
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -12,7 +12,7 @@ from smolagents import (
12
  from smolagents.models import LiteLLMModel
13
  #from smolagents.prompts import CODE_SYSTEM_PROMPT
14
  from huggingface_hub import login
15
- from dotenv import load_dotenv
16
  from bs4 import BeautifulSoup
17
  import json
18
  import time
@@ -20,7 +20,7 @@ from datetime import datetime, timedelta
20
  from collections import deque
21
 
22
  # Load environment variables and authenticate
23
- load_dotenv()
24
  hf_token = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
25
  if hf_token:
26
  login(token=hf_token)
@@ -359,7 +359,7 @@ class FinalAgent:
359
  self.model = LiteLLMModel(
360
  api_key=api_key,
361
  # model_id="gpt-4o",
362
- model_id="vertex_ai/gemini-pro",
363
  # model_id="anthropic/claude-3-5-sonnet-20240620",
364
  temperature=0.5 # Added temperature parameter
365
  )
 
12
  from smolagents.models import LiteLLMModel
13
  #from smolagents.prompts import CODE_SYSTEM_PROMPT
14
  from huggingface_hub import login
15
+ #from dotenv import load_dotenv --- don't need HF spaces
16
  from bs4 import BeautifulSoup
17
  import json
18
  import time
 
20
  from collections import deque
21
 
22
  # Load environment variables and authenticate
23
+ #load_dotenv() --- see above. Environment variables are loaded in HF Space -> Settings -> Secrets
24
  hf_token = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
25
  if hf_token:
26
  login(token=hf_token)
 
359
  self.model = LiteLLMModel(
360
  api_key=api_key,
361
  # model_id="gpt-4o",
362
+ model_id="gemini/gemini-pro",
363
  # model_id="anthropic/claude-3-5-sonnet-20240620",
364
  temperature=0.5 # Added temperature parameter
365
  )