c-ho commited on
Commit
7599b35
·
verified ·
1 Parent(s): ff831d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -40,6 +40,16 @@ def display_model_info(model_name):
40
  link = f"[Open model page]({info['link']})"
41
  return usage_code, link
42
 
 
 
 
 
 
 
 
 
 
 
43
 
44
  # -----------------------
45
  # NER function (SAFE OUTPUT)
 
40
  link = f"[Open model page]({info['link']})"
41
  return usage_code, link
42
 
43
+ model_cache = {}
44
+ def get_model(model_name):
45
+ if model_name not in model_cache:
46
+ model_cache[model_name] = pipeline(
47
+ "ner",
48
+ model=model_name,
49
+ aggregation_strategy="simple" # safer for display
50
+ )
51
+ return model_cache[model_name]
52
+
53
 
54
  # -----------------------
55
  # NER function (SAFE OUTPUT)