Update app.py

app.py

@@ -1,12 +1,12 @@
 import os
-os.environ["TRANSFORMERS_NO_TF"] = "1"
+#os.environ["TRANSFORMERS_NO_TF"] = "1"
 
 import gradio as gr
 import requests
 import inspect
 import pandas as pd
 import time
-from transformers import pipeline
+#from transformers import pipeline
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
 # (Keep Constants as is)
 # --- Constants ---
@@ -19,7 +19,7 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         # Initialize a small local model instead of API model
-        self.model = pipeline("text-generation", model="tiiuae/falcon-rw-1b")
+        self.model = HfApiModel("text-generation", model="mistralai/Mistral-7B-v0.1")
 
         # Initialize the search tool (you can keep DuckDuckGoSearchTool)
         search_tool = DuckDuckGoSearchTool()