lopera47 commited on
Commit
b4fdf61
·
verified ·
1 Parent(s): 518a7d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -35,7 +35,15 @@ class BasicAgent:
35
  print("BasicAgent initialized.")
36
 
37
  # Initialize the Agent
38
- self.agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=InferenceClientModel())
 
 
 
 
 
 
 
 
39
 
40
  def __call__(self, question: str) -> str:
41
  print(f"Agent received question (first 50 chars): {question[:50]}...")
 
35
  print("BasicAgent initialized.")
36
 
37
  # Initialize the Agent
38
+ model = InferenceClientModel(
39
+ model_id="tiiuae/falcon-7b-instruct",
40
+ token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
41
+ )
42
+
43
+ self.agent = CodeAgent(
44
+ tools=[DuckDuckGoSearchTool()],
45
+ model=model
46
+ )
47
 
48
  def __call__(self, question: str) -> str:
49
  print(f"Agent received question (first 50 chars): {question[:50]}...")