alyxx
commited on
Commit
·
b1a6713
1
Parent(s):
cb77f6c
update app.py and requirements
Browse files
app.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
|
| 2 |
import gradio as gr
|
| 3 |
|
| 4 |
# Load the NER pipeline
|
|
@@ -18,7 +18,7 @@ def fn_ner(prompt):
|
|
| 18 |
examples = [
|
| 19 |
"Subject: Defective Date: 08-13-2023 Product: XXX speaker Location: 456 Sound Avenue, Audiotown",
|
| 20 |
"Subject: Dirty Date: 08-10-2023 Product: UVW Television Location: 567 Willow Lane, Mediatown",
|
| 21 |
-
"Subject: Missing Parts Date: 08-10-2023 Product: XXX Furniture Set Location:
|
| 22 |
]
|
| 23 |
|
| 24 |
# Create Gradio interface
|
|
@@ -28,7 +28,7 @@ iface = gr.Interface(
|
|
| 28 |
outputs='text',
|
| 29 |
examples=[[ex] for ex in examples],
|
| 30 |
title='Named Entity Recognition',
|
| 31 |
-
description='This demo performs named entity recognition (NER) on
|
| 32 |
article='All done by Kaiku'
|
| 33 |
)
|
| 34 |
|
|
|
|
| 1 |
+
ifrom transformers import pipeline
|
| 2 |
import gradio as gr
|
| 3 |
|
| 4 |
# Load the NER pipeline
|
|
|
|
| 18 |
examples = [
|
| 19 |
"Subject: Defective Date: 08-13-2023 Product: XXX speaker Location: 456 Sound Avenue, Audiotown",
|
| 20 |
"Subject: Dirty Date: 08-10-2023 Product: UVW Television Location: 567 Willow Lane, Mediatown",
|
| 21 |
+
"Subject: Missing Parts Date: 08-10-2023 Product: XXX Furniture Set Location: Antipolo Rizal",
|
| 22 |
]
|
| 23 |
|
| 24 |
# Create Gradio interface
|
|
|
|
| 28 |
outputs='text',
|
| 29 |
examples=[[ex] for ex in examples],
|
| 30 |
title='Named Entity Recognition',
|
| 31 |
+
description='This demo performs named entity recognition (NER) on our custom made dataset. This dataset consist of very small training and testing samples resulting a very limited data our model to learn. ',
|
| 32 |
article='All done by Kaiku'
|
| 33 |
)
|
| 34 |
|