Update app.py

app.py

@@ -3,6 +3,7 @@ from pyvis.network import Network
 import networkx as nx
 import numpy as np
 import pandas as pd
+import os
 from datasets import load_dataset