ATK20 commited on
Commit
e7b5b0a
·
verified ·
1 Parent(s): 1bef83e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ class BasicAgent:
19
  def __init__(self):
20
  print("BasicAgent initialized.")
21
  # Initialize a small local model instead of API model
22
- self.model = pipeline("text-generation", model="tiiuae/falcon-rw-1b ")
23
 
24
  # Initialize the search tool (you can keep DuckDuckGoSearchTool)
25
  search_tool = DuckDuckGoSearchTool()
 
19
  def __init__(self):
20
  print("BasicAgent initialized.")
21
  # Initialize a small local model instead of API model
22
+ self.model = pipeline("text-generation", model="tiiuae/falcon-rw-1b")
23
 
24
  # Initialize the search tool (you can keep DuckDuckGoSearchTool)
25
  search_tool = DuckDuckGoSearchTool()