Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|