Adds en_core_web_sm model
Browse files
app.py
CHANGED
|
@@ -11,6 +11,7 @@ subprocess.run(["pip", "install", "./GLiREL"])
|
|
| 11 |
subprocess.run(["pip", "install", "-r", "./GLiREL/requirements.txt"])
|
| 12 |
|
| 13 |
subprocess.run(["pip", "install", "spacy"])
|
|
|
|
| 14 |
subprocess.run(["python", "-m", "spacy", "download", "en_core_web_lg"])
|
| 15 |
|
| 16 |
from typing import Dict, Union
|
|
@@ -92,7 +93,7 @@ with gr.Blocks() as demo:
|
|
| 92 |
gr.Markdown("GitHub: https://github.com/jackboyla/GLiREL")
|
| 93 |
|
| 94 |
text_input = gr.Textbox(label="Input Text", value="Jack lives in London but he was born in Mongolia.")
|
| 95 |
-
model_name_input = gr.Dropdown(choices=["en_core_web_lg"], label="NER Model", value="en_core_web_lg")
|
| 96 |
labels_input = gr.Textbox(label="Relation Labels (comma-separated)", value="country of origin, licensed to broadcast to, father, followed by, characters")
|
| 97 |
|
| 98 |
ner_output = gr.HighlightedText(label="NER")
|
|
|
|
| 11 |
subprocess.run(["pip", "install", "-r", "./GLiREL/requirements.txt"])
|
| 12 |
|
| 13 |
subprocess.run(["pip", "install", "spacy"])
|
| 14 |
+
subprocess.run(["python", "-m", "spacy", "download", "en_core_web_sm"])
|
| 15 |
subprocess.run(["python", "-m", "spacy", "download", "en_core_web_lg"])
|
| 16 |
|
| 17 |
from typing import Dict, Union
|
|
|
|
| 93 |
gr.Markdown("GitHub: https://github.com/jackboyla/GLiREL")
|
| 94 |
|
| 95 |
text_input = gr.Textbox(label="Input Text", value="Jack lives in London but he was born in Mongolia.")
|
| 96 |
+
model_name_input = gr.Dropdown(choices=["en_core_web_sm", "en_core_web_lg"], label="NER Model", value="en_core_web_lg")
|
| 97 |
labels_input = gr.Textbox(label="Relation Labels (comma-separated)", value="country of origin, licensed to broadcast to, father, followed by, characters")
|
| 98 |
|
| 99 |
ner_output = gr.HighlightedText(label="NER")
|