nickprock commited on
Commit
f828f91
1 Parent(s): 530fd26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -1
app.py CHANGED
@@ -1,3 +1,24 @@
 
 
 
1
  import gradio as gr
2
 
3
- gr.Interface.load("models/nickprock/bert-italian-finetuned-ner").launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ from transformers import pipeline
3
+
4
  import gradio as gr
5
 
6
+ ner_pipeline = pipeline("ner", model="nickprock/bert-italian-finetuned-ner")
7
+
8
+ examples = [
9
+ ["Domani andrò allo stadio con Giovanna a vedere la Fiorentina"],
10
+ ["La sede storica della Olivetti è ad Ivrea"],
11
+ ["Ieri sera c'è stato Harry Potter in TV"]
12
+
13
+ ]
14
+
15
+ def ner(text):
16
+ output = ner_pipeline(text)
17
+ return {"text": text, "entities": output}
18
+
19
+ demo = gr.Interface(ner,
20
+ gr.Textbox(placeholder="Inserisci una frase qui..."),
21
+ gr.HighlightedText(),
22
+ examples=examples)
23
+
24
+ demo.launch()