Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from datasets import load_dataset, concatenate_datasets, save_dataset
+from datasets import load_dataset, concatenate_datasets
 
 def load_and_combine_datasets():
     python_codes_dataset = load_dataset('flytech/python-codes-25k', split='train')
@@ -11,7 +11,7 @@ def load_and_combine_datasets():
     return combined_dataset
 
 def save_combined_dataset(combined_dataset, dataset_name):
-    save_dataset(combined_dataset, dataset_name)
+    combined_dataset.save_to_disk(dataset_name)
 
 def main():
     st.title("Combined Dataset Viewer")