Update app.py

app.py

@@ -3,6 +3,7 @@ import time
 import json
 from gensim.models import Word2Vec
 import pandas as pd
+from datasets import load_dataset
 
 # Define the HTML and CSS styles
 html_temp = """
@@ -90,6 +91,12 @@ if query:
     st.header(f"Similar Genes to {query}")
     st.write(df1.head(50))
 
+    # for split, data in dataset.items():
+    # data.to_csv(f"my-dataset-{split}.csv", index = None)
+
+    # for split, data in dataset.items():
+    # data.to_csv(f"my-dataset-{split}.js", index = None)
+
     # CSV format
     dataset = df1.head(50)
     dataset.to_csv("sim2.csv", index=None)