Update app.py

app.py

@@ -46,8 +46,8 @@ elif page == "Model Evaluation":
     # Load dataset
     with st.spinner("Loading dataset..."):
         dataset = load_dataset(dataset_name, task, split=split)
-
-    st.write(f"Loaded {len(dataset)} samples from {dataset_name} ({task}/{split}).")
+        dataset = dataset.select(range(1000))  # Limit to first 1000 samples
+        st.write(f"Loaded {len(dataset)} samples from {dataset_name} ({task}/{split}).")
 
     # Load model
     model_name = f"cardiffnlp/twitter-roberta-base-{task}"
@@ -91,3 +91,4 @@ elif page == "Model Evaluation":
 
     st.pyplot(fig)
     st.success(f"Precision-Recall AUC: {pr_auc:.2f}")
+