Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
103 |
-
if title and 'textile' in title.lower():
|
104 |
-
return title
|
105 |
-
attempts += 1
|
106 |
except StopIteration:
|
107 |
break
|
108 |
-
|
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)
|