Update app.py

app.py

@@ -22,14 +22,14 @@ if "df" not in st.session_state:
     st.session_state.df = None
 
 # Dataset loading section
-#st.subheader("Load Dataset")
+st.subheader("Load Dataset")
 input_option = st.radio("Select Dataset Input:", ["Use Hugging Face Dataset", "Upload CSV File"])
 
 if input_option == "Use Hugging Face Dataset":
     dataset_name = st.text_input("Enter Hugging Face Dataset Name:", value="HUPD/hupd")
     if st.button("Load Dataset"):
         try:
-            # Ensure `uniform_split=True` is provided for compatibility
+            # Load dataset and store it in session state
             dataset = load_dataset(dataset_name, name="sample", split="train", trust_remote_code=True, uniform_split=True)
             st.session_state.df = pd.DataFrame(dataset)
             st.success(f"Dataset '{dataset_name}' loaded successfully!")
@@ -37,8 +37,9 @@ if input_option == "Use Hugging Face Dataset":
             st.error(f"Error loading dataset: {e}")
 elif input_option == "Upload CSV File":
     uploaded_file = st.file_uploader("Upload CSV File:", type=["csv"])
-    if uploaded_file:
+    if uploaded_file and st.button("Load CSV"):
         try:
+            # Read uploaded CSV and store it in session state
             st.session_state.df = pd.read_csv(uploaded_file)
             st.success("File uploaded successfully!")
         except Exception as e: