cyberandy commited on
Commit
a168d09
·
1 Parent(s): 40806d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -5,10 +5,6 @@ import requests
5
  import json
6
  import spacy
7
 
8
- # Load German model
9
- nlp_model_de = spacy.load("de_core_news_lg")
10
- nlp_model_de.add_pipe("entityfishing")
11
-
12
  # Page config
13
  st.set_page_config(
14
  page_title="Entity Linking by WordLift",
@@ -43,7 +39,7 @@ def load_model(selected_language, model_name=None, entity_set=None):
43
  if selected_language == "German":
44
  # Load the German-specific model
45
  nlp_model_de = spacy.load("de_core_news_lg")
46
- nlp_model_de.add_pipe("entityfishing", config={"language": "de"})
47
 
48
  return nlp_model_de
49
  else:
 
5
  import json
6
  import spacy
7
 
 
 
 
 
8
  # Page config
9
  st.set_page_config(
10
  page_title="Entity Linking by WordLift",
 
39
  if selected_language == "German":
40
  # Load the German-specific model
41
  nlp_model_de = spacy.load("de_core_news_lg")
42
+ nlp_model_de.add_pipe("entityfishing")
43
 
44
  return nlp_model_de
45
  else: