Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from datasets import load_dataset
+from datasets import load_dataset, Dataset
 from PIL import Image
 import io
 import base64
@@ -17,6 +17,9 @@ dataset = dataset['_data_files']
 
 print(f'This is the train dataset: {dataset}')
 
+dataset = Dataset.from_file(dataset[0][0]['filename'])
+
+
 def reshape_json_data_to_fit_visualize_graph(graph_data):
 
     nodes = graph_data["nodes"]