Tanor commited on
Commit
3dc6f15
1 Parent(s): 7810ff4
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -5,26 +5,25 @@ from spacy import displacy
5
  nlp = spacy.load("sr_ner_tesla_j125")
6
 
7
  def calculate(text):
8
- doc = nlp(text)
9
- return displacy.render(doc, style="ent")
 
 
 
10
 
11
  iface = gr.Interface(
12
  fn=calculate,
13
  inputs=gr.Textbox(lines=5, placeholder="Unesite rečenicu ovde..."),
14
  outputs=gr.HTML(label="Imenovani entiteti"),
15
  title="NER for Serbian",
16
- description="""
17
- Demostacija spaCy NER modela sa jerteh125 BERT-om.
18
- """,
19
  examples=[
20
  ["Ana voli Milovana."],
21
  ["Brograd je glavni grad Srbije."],
22
  ["Nikoli Tesla je bio izumitelj i naučnik."],
23
  ["Monaliza je slika Leonarda da Vinčija."],
24
  ["Sveti Sava je bio prvi srpski arhiepiskop."],
25
-
26
  ]
27
-
28
  )
29
 
30
  iface.launch()
 
5
  nlp = spacy.load("sr_ner_tesla_j125")
6
 
7
  def calculate(text):
8
+ try:
9
+ doc = nlp(text)
10
+ return displacy.render(doc, style="ent")
11
+ except Exception as e:
12
+ return f"Error: {e}"
13
 
14
  iface = gr.Interface(
15
  fn=calculate,
16
  inputs=gr.Textbox(lines=5, placeholder="Unesite rečenicu ovde..."),
17
  outputs=gr.HTML(label="Imenovani entiteti"),
18
  title="NER for Serbian",
19
+ description="Demonstracija spaCy NER modela sa jerteh125 BERT-om.",
 
 
20
  examples=[
21
  ["Ana voli Milovana."],
22
  ["Brograd je glavni grad Srbije."],
23
  ["Nikoli Tesla je bio izumitelj i naučnik."],
24
  ["Monaliza je slika Leonarda da Vinčija."],
25
  ["Sveti Sava je bio prvi srpski arhiepiskop."],
 
26
  ]
 
27
  )
28
 
29
  iface.launch()