max-long commited on
Commit
c09c624
·
verified ·
1 Parent(s): 731c8b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -92,20 +92,17 @@ with gr.Blocks(title="Textile Machinery NER Demo") as demo:
92
 
93
  submit_btn = gr.Button("Analyze Random Snippet")
94
  refresh_btn = gr.Button("Get New Snippet")
95
-
96
  def get_new_snippet():
97
  attempts = 0
98
  max_attempts = 1000 # Prevent infinite loops
99
  while attempts < max_attempts:
100
  try:
101
- sample = next(dataset_iter)
102
- title = sample.get('title', '')
103
- if title and 'textile' in title.lower():
104
- return title
105
- attempts += 1
106
  except StopIteration:
107
  break
108
- return "No more snippets available."
109
 
110
  # Connect refresh button
111
  refresh_btn.click(fn=get_new_snippet, outputs=input_text)
 
92
 
93
  submit_btn = gr.Button("Analyze Random Snippet")
94
  refresh_btn = gr.Button("Get New Snippet")
95
+
96
  def get_new_snippet():
97
  attempts = 0
98
  max_attempts = 1000 # Prevent infinite loops
99
  while attempts < max_attempts:
100
  try:
101
+ sample = next(dataset_iter)['text']
102
+ return sample
 
 
 
103
  except StopIteration:
104
  break
105
+ return "No more snippets available."
106
 
107
  # Connect refresh button
108
  refresh_btn.click(fn=get_new_snippet, outputs=input_text)