infinitymatter commited on
Commit
c6201fd
·
verified ·
1 Parent(s): a103ecb

Update src/models.py

Browse files
Files changed (1) hide show
  1. src/models.py +3 -0
src/models.py CHANGED
@@ -8,6 +8,9 @@ from huggingface_hub import InferenceClient
8
  # Load environment variables from .env file
9
  load_dotenv(override=True)
10
 
 
 
 
11
 
12
  # Initialize disguised API clients
13
  openai = InferenceClient(model="mistralai/Mistral-7B-Instruct-v0.3", token=openai_api_key)
 
8
  # Load environment variables from .env file
9
  load_dotenv(override=True)
10
 
11
+ # Retrieve API keys from environment
12
+ openai_api_key = os.getenv("OPENAI_API_KEY")
13
+ anthropic_api_key = os.getenv("ANTHROPIC_API_KEY")
14
 
15
  # Initialize disguised API clients
16
  openai = InferenceClient(model="mistralai/Mistral-7B-Instruct-v0.3", token=openai_api_key)