Update app.py

app.py

@@ -101,8 +101,9 @@ with gr.Blocks(title="Textile Machinery NER Demo") as demo:
                 sample = next(dataset_iter)['text']
                 return sample
             except StopIteration:
-                break
-            return "No more snippets available."   
+                break  # Exit the loop if we run out of snippets
+            attempts += 1  # Increment attempts
+        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)