Update app.py

app.py

@@ -93,8 +93,8 @@ if query:
     st.write(df1.head(50))
 
 
-    arrow_dataset = load_dataset(df1.head(50))
-    arrow_dataset.save_to_disk("sim2.js")
+    arrow_dataset = Dataset.from_pandas(df1.head(50))
+    arrow_dataset.save_to_disk("word2vec_dataset")
 
     # arrow_dataset_reloaded = load_from_disk('sim2.js')
     # arrow_dataset_reloaded