Update app.py

app.py

@@ -94,18 +94,18 @@ with gr.Blocks(title="Textile Machinery NER Demo") as demo:
     refresh_btn = gr.Button("Get New Snippet")
     
     def get_new_snippet():
-    attempts = 0
-    max_attempts = 1000  # Prevent infinite loops
-    while attempts < max_attempts:
-        try:
-            sample = next(dataset_iter)
-            title = sample.get('title', '')
-            if title and 'textile' in title.lower():
-                return title
-            attempts += 1
-        except StopIteration:
-            break
-    return "No more snippets available."
+        attempts = 0
+        max_attempts = 1000  # Prevent infinite loops
+        while attempts < max_attempts:
+            try:
+                sample = next(dataset_iter)
+                title = sample.get('title', '')
+                if title and 'textile' in title.lower():
+                    return title
+                attempts += 1
+            except StopIteration:
+                break
+                return "No more snippets available."
     
     # Connect refresh button
     refresh_btn.click(fn=get_new_snippet, outputs=input_text)