Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ import json
|
|
| 9 |
from transformers import AutoTokenizer, AutoModelForCausalLM , pipeline, TransformersEngine
|
| 10 |
import wikipedia
|
| 11 |
import torch
|
| 12 |
-
from tooling import ModelMathTool,
|
| 13 |
|
| 14 |
# (Keep Constants as is)
|
| 15 |
# --- Constants ---
|
|
@@ -26,7 +26,7 @@ cache= {}
|
|
| 26 |
web_search = DuckDuckGoSearchTool()
|
| 27 |
python_interpreter = PythonInterpreterTool()
|
| 28 |
visit_webpage_tool = VisitWebpageTool()
|
| 29 |
-
wiki_tool =
|
| 30 |
|
| 31 |
#model_math_tool = ModelMathTool()
|
| 32 |
|
|
|
|
| 9 |
from transformers import AutoTokenizer, AutoModelForCausalLM , pipeline, TransformersEngine
|
| 10 |
import wikipedia
|
| 11 |
import torch
|
| 12 |
+
from tooling import ModelMathTool, WikipediaTool
|
| 13 |
|
| 14 |
# (Keep Constants as is)
|
| 15 |
# --- Constants ---
|
|
|
|
| 26 |
web_search = DuckDuckGoSearchTool()
|
| 27 |
python_interpreter = PythonInterpreterTool()
|
| 28 |
visit_webpage_tool = VisitWebpageTool()
|
| 29 |
+
wiki_tool = WikipediaTool()
|
| 30 |
|
| 31 |
#model_math_tool = ModelMathTool()
|
| 32 |
|