Update app.py

app.py

@@ -71,5 +71,16 @@ def find_samples_to_label():
     top_pred_confs = 1 - np.max(preds, axis=1)
     idx_to_label = np.argpartition(top_pred_confs, -5)[-5:]
 
+    # Upload samples to the dataset to label
     to_label_data = unlabeled_data.select(idx_to_label)
-    to_label_data.push_to_hub("active-learning/to_label_samples")
+    to_label_data.push_to_hub("active-learning/to_label_samples")
+
+    # Remove from the pool of samples
+    unlabeled_data = unlabeled_data.select(
+        (
+            i for i in range(len(unlabeled_data)) 
+            if i not in set(idx_to_label)
+        )
+    )
+    unlabeled_data.push_to_hub("active-learning/unlabeled_samples")
+