dwb2023 commited on
Commit
32e2d9a
·
verified ·
1 Parent(s): d2d2f97

add pyvis to app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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