Update app.py

app.py

@@ -7,14 +7,12 @@ from datasets import load_dataset
 
 # Hugging Face Datasets
 @st.cache_data
-
 def load_data():
-    network_insights = load_dataset("huggingface/infrastructure-insights", split="train")
-    broadband_forecast = load_dataset("huggingface/broadband-access-forecast", split="train")
-    return network_insights.to_pandas(), broadband_forecast.to_pandas()
+    network_insights = load_dataset("infinite-dataset-hub/5GNetworkOptimization", split="train")
+    return network_insights.to_pandas()
 
 # Load Datasets
-network_insights, broadband_forecast = load_data()
+network_insights = load_data()
 
 # Title
 st.title("Smart Network Infrastructure Planner")
@@ -26,15 +24,12 @@ priority_area = st.sidebar.selectbox("Priority Area:", ["Rural", "Urban", "Subur
 signal_threshold = st.sidebar.slider("Signal Strength Threshold (dBm):", min_value=-120, max_value=-30, value=-80)
 
 # Display Dataset Options
-data_to_view = st.sidebar.selectbox("Select Dataset to View:", ["Network Insights", "Broadband Forecast"])
+data_to_view = st.sidebar.selectbox("Select Dataset to View:", ["Network Insights"])
 
 # Display Selected Dataset
 if data_to_view == "Network Insights":
     st.subheader("Network Insights Dataset")
     st.dataframe(network_insights)
-else:
-    st.subheader("Broadband Forecast Dataset")
-    st.dataframe(broadband_forecast)
 
 # Terrain and Connectivity Analysis Section
 st.header("Terrain and Connectivity Analysis")