Update app.py

app.py

@@ -13,6 +13,7 @@ import urllib.request
 import random
 import plotly.express as px
 
+
 st.set_page_config(
     page_title="Abstractalytics",
                 page_icon=":microscope:",
@@ -926,11 +927,13 @@ if query:
     st.markdown("---")
 
 
+    import os
+    
     from datasets import Dataset
 
     # Load existing comments or create a new dataset
-    if Dataset.get_dataset_infos('comments'):
-        dataset = Dataset.load_from_disk('comments')
+    if os.path.exists('comments'):
+    dataset = Dataset.load_from_disk('comments')
     else:
         dataset = Dataset.from_dict({'id': [], 'text': []})
 
@@ -942,7 +945,7 @@ if query:
         # Save the dataset to disk
         dataset.save_to_disk('comments')
 
-    print('Comment saved to dataset.')
+        print('Comment saved to dataset.')
 
     st.title("Abstractalytics Web App")
     st.write("We appreciate your feedback!")