Update app.py

app.py

@@ -66,8 +66,17 @@ for i in range(1, hidden_layers + 1):
         st.error(f"Invalid input for the number of neurons in Layer {i}. Please enter an integer.")
 
 # Batch Size
-if data_set != "None":
-    df = pd.read_csv('data_set')
+if data_set:
+    try:
+        # Read the CSV file using the provided file name
+        df = pd.read_csv(data_set)
+        
+        # Display the DataFrame
+        st.write(df)
+    except FileNotFoundError:
+        st.error(f"File '{data_set}' not found. Please check the file name and try again.")
+    except Exception as e:
+        st.error(f"An error occurred: {e}")
     X = df.iloc[:, :2].values
     y = df.iloc[:, -1].values
     batch_size = st.sidebar.select_slider("Batch Size", options=[i for i in range(1, X.shape[0] + 1)])