Update app.py

app.py

@@ -81,13 +81,16 @@ with gr.Blocks(title="General NER with Color-Coded Output") as demo:
 
     def get_new_snippet():
         # Preload several samples into a list
-        samples = [sample['text'] for sample, _ in zip(dataset_iter, range(100))]  # Load 100 samples
+        max_length = 384  # Maximum length for snippets
+        samples = [
+            sample['text'][:max_length] for sample, _ in zip(dataset_iter, range(100))  # Truncate to max_length
+        ]
         
         # 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  
+        return "No more snippets available."  # Return this if no valid snippets are found 
     
     # Connect refresh button
     refresh_btn.click(fn=get_new_snippet, outputs=input_text)