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")
|