Update app.py

app.py

@@ -79,12 +79,14 @@ with gr.Blocks(title="General NER with Color-Coded Output") as demo:
     output_entities = gr.JSON(label="Entities")
     
 
-
     def get_new_snippet():
-        attempts = 0
-        max_attempts = 1000  # Prevent infinite loops
-        for sample in dataset_iter:
-            return sample['text']
+        # Preload several samples into a list
+        samples = [sample['text'] for sample, _ in zip(dataset_iter, range(100))]  # Load 100 samples
+        
+        # Return a random snippet from the preloaded samples
+        if samples:
+            return random.choice(samples)
+        
         return "No more snippets available."  # Return this if no valid snippets are found  
     
     # Connect refresh button