Update app.py

app.py

@@ -93,13 +93,19 @@ with gr.Blocks(title="Textile Machinery NER Demo") as demo:
     submit_btn = gr.Button("Analyze Random Snippet")
     refresh_btn = gr.Button("Get New Snippet")
     
-    # Function to fetch a new random snippet
     def get_new_snippet():
+    attempts = 0
+    max_attempts = 1000  # Prevent infinite loops
+    while attempts < max_attempts:
         try:
-            sample = next(dataset_iter)['text']
-            return sample
+            sample = next(dataset_iter)
+            title = sample.get('title', '')
+            if title and 'textile' in title.lower():
+                return title
+            attempts += 1
         except StopIteration:
-            return "No more snippets available."
+            break
+    return "No more snippets available."
     
     # Connect refresh button
     refresh_btn.click(fn=get_new_snippet, outputs=input_text)