Spaces:

MADtoBAD
/

my_agent

Runtime error

MADtoBAD commited on Oct 24

Commit

d29a348

verified ·

1 Parent(s): 07752af

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from smolagents.models import TransformersModel
 class SimpleAIAgent:
     def __init__(self):
         print("Initializing AI Agent...")
-        self.model = TransformersModel("microsoft/DialoGPT-medium")
         self.search_tool = DuckDuckGoSearchTool()

 class SimpleAIAgent:
     def __init__(self):
         print("Initializing AI Agent...")
+        self.model = TransformersModel("microsoft/DialoGPT-small")
         self.search_tool = DuckDuckGoSearchTool()