jackboyla commited on
Commit
1cc3461
·
1 Parent(s): 58ff769

Adds en_core_web_sm model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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")