crabz commited on
Commit
1ef1ab1
1 Parent(s): 86433c4

print no labels if subword labels are found or exception is thrown

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -1,8 +1,8 @@
1
  # -*- coding: utf-8 -*-
2
  import os
3
 
4
- os.system("pip3 install torch==1.10.1+cpu torchvision==0.11.2+cpu torchaudio==0.10.1+cpu -f "
5
- "https://download.pytorch.org/whl/cpu/torch_stable.html")
6
 
7
  import gradio as gr
8
  from transformers import pipeline
@@ -70,7 +70,10 @@ def set_entities(sentence, entities):
70
  ents = []
71
  for label, start, end in entities:
72
  ents.append(doc.char_span(start, end, label))
73
- doc.ents = ents
 
 
 
74
  return doc
75
 
76
 
@@ -100,5 +103,6 @@ intf = gr.Interface(fn=apply_ner, inputs="text", outputs="html", title='Slovak N
100
  "<li><em>Memorial založili v roku 1987 disidenti na čele s laureátom Nobelovej ceny za mier akademikom Andrejom Sacharovom.</em> (year classified as location)</li>"
101
  "<li><em>Išla som do Petrozavodska navštíviť vo väzbe Jurija Dmitrijeva, vedúceho karelskej pobočky Memorialu.</em> (<em>Memorialu</em> classified as location instead of organization)</li></ul>"
102
  "Year is often missclassified as location, this could be fixed by introducing new time label.<br/>"
103
- "Location and organization can be difficult to distinguish in some cases, because both are suitable. Is Internation Space Station organization or location?")
 
104
  intf.launch()
 
1
  # -*- coding: utf-8 -*-
2
  import os
3
 
4
+ # os.system("pip3 install torch==1.10.1+cpu torchvision==0.11.2+cpu torchaudio==0.10.1+cpu -f "
5
+ # "https://download.pytorch.org/whl/cpu/torch_stable.html")
6
 
7
  import gradio as gr
8
  from transformers import pipeline
 
70
  ents = []
71
  for label, start, end in entities:
72
  ents.append(doc.char_span(start, end, label))
73
+ try:
74
+ doc.ents = ents
75
+ except TypeError:
76
+ pass
77
  return doc
78
 
79
 
 
103
  "<li><em>Memorial založili v roku 1987 disidenti na čele s laureátom Nobelovej ceny za mier akademikom Andrejom Sacharovom.</em> (year classified as location)</li>"
104
  "<li><em>Išla som do Petrozavodska navštíviť vo väzbe Jurija Dmitrijeva, vedúceho karelskej pobočky Memorialu.</em> (<em>Memorialu</em> classified as location instead of organization)</li></ul>"
105
  "Year is often missclassified as location, this could be fixed by introducing new time label.<br/>"
106
+ "Location and organization can be difficult to distinguish in some cases, because both are suitable. Is Internation Space Station organization or location?",
107
+ analytics_enabled=False)
108
  intf.launch()