Update src/models.py

src/models.py

@@ -2,6 +2,7 @@ from openai import OpenAI
 import anthropic
 import os
 from dotenv import load_dotenv
+from huggingface_hub import HfApi;
 
 # Load environment variables from .env file
 load_dotenv(override=True)
@@ -18,12 +19,12 @@ if not anthropic_api_key:
     print("❌ Anthropic API Key is missing!")
 
 # Initialize API clients
-openai = OpenAI(api_key=openai_api_key)
-claude = anthropic.Anthropic()
+openai = HfApi(token=openai_api_key)
+claude = HfApi(token=anthropic_api_key)
 
 # Model names
-OPENAI_MODEL = "gpt-4o-mini"
-CLAUDE_MODEL = "claude-3-5-sonnet-20240620"
+OPENAI_MODEL = "mistralai/Mistral-7B-Instruct-v0.3"
+CLAUDE_MODEL = "mistralai/Mistral-7B-Instruct-v0.3"
 
 # Call OpenAI's GPT model with prompt and system message
 def get_gpt_completion(prompt, system_message):