Update app.py

app.py

@@ -5,6 +5,7 @@ import inspect
 import pandas as pd
 from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool
 from tools import TextTransformerTool
+from smolagents import InferenceClientModel
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -18,7 +19,6 @@ import os
 from smolagents import HfApiModel
 
 
-
 class BasicAgent:
     def __init__(self):
         print("Initializing BasicAgent with tools...")
@@ -27,9 +27,10 @@ class BasicAgent:
         hf_token = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
         if not hf_token:
             raise ValueError("Missing Hugging Face API token!")
-        
+        openai_token = os.environ.get("OPENAI_API_KEY")
         # Pass it into the model
-        model = HfApiModel(model="mistralai/Mistral-7B-Instruct-v0.2") #token=hf_token,    
+        # model = HfApiModel(model="mistralai/Mistral-7B-Instruct-v0.2") #token=hf_token,
+        model = InferenceClientModel(provider="openai", token=openai_token)
         search_tool = DuckDuckGoSearchTool()
         text_tool = TextTransformerTool()