Spaces:
Sleeping
Sleeping
add pyvis to app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
import os
|
2 |
import networkx as nx
|
|
|
3 |
from smolagents import CodeAgent, HfApiModel, tool, GradioUI
|
4 |
|
5 |
# Define a tool for analyzing the Florentine Families graph
|
@@ -35,7 +36,7 @@ model = HfApiModel()
|
|
35 |
agent = CodeAgent(
|
36 |
tools=[analyze_florentine_graph],
|
37 |
model=model,
|
38 |
-
additional_authorized_imports=["networkx"],
|
39 |
add_base_tools=True
|
40 |
)
|
41 |
|
|
|
1 |
import os
|
2 |
import networkx as nx
|
3 |
+
import pyvis
|
4 |
from smolagents import CodeAgent, HfApiModel, tool, GradioUI
|
5 |
|
6 |
# Define a tool for analyzing the Florentine Families graph
|
|
|
36 |
agent = CodeAgent(
|
37 |
tools=[analyze_florentine_graph],
|
38 |
model=model,
|
39 |
+
additional_authorized_imports=["networkx","pyvis"],
|
40 |
add_base_tools=True
|
41 |
)
|
42 |
|