Update app.py

app.py

@@ -2,26 +2,20 @@ import streamlit as st
 from datasets import load_dataset, concatenate_datasets
 
 def load_and_combine_datasets():
-    # Load the datasets
     python_codes_dataset = load_dataset('flytech/python-codes-25k', split='train')
-    streamlit_issues_dataset = load_dataset("andfanilo/streamlit-issues")
-    streamlit_docs_dataset = load_dataset("sai-lohith/streamlit_docs")
-
-    # Combine the datasets
+    streamlit_issues_dataset = load_dataset("andfanilo/streamlit-issues", split='train')
+    streamlit_docs_dataset = load_dataset("sai-lohith/streamlit_docs", split='train')
     combined_dataset = concatenate_datasets([python_codes_dataset, streamlit_issues_dataset, streamlit_docs_dataset])
     return combined_dataset
 
 def main():
     st.title("Combined Dataset Viewer")
-    
+
     # Load and combine datasets
     combined_dataset = load_and_combine_datasets()
-    
-    # Display random sample from the combined dataset
-    random_sample = combined_dataset.shuffle(seed=42).select(range(10))
-    
-    st.header("Random Sample from Combined Dataset")
-    st.write(random_sample)
+
+    # Display combined dataset
+    st.write("Combined Dataset:", combined_dataset)
 
 if __name__ == "__main__":
     main()