Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +1 -1
src/streamlit_app.py
CHANGED
|
@@ -127,7 +127,7 @@ def load_gliner_model(model_name):
|
|
| 127 |
"""Initializes and caches the GLiNER model."""
|
| 128 |
try:
|
| 129 |
if model_name == "HR_AI":
|
| 130 |
-
return GLiNER.from_pretrained("
|
| 131 |
elif model_name == "InfoFinder":
|
| 132 |
return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.5", device="cpu")
|
| 133 |
except Exception as e:
|
|
|
|
| 127 |
"""Initializes and caches the GLiNER model."""
|
| 128 |
try:
|
| 129 |
if model_name == "HR_AI":
|
| 130 |
+
return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.5", nested_ner=True, num_gen_sequences=2, gen_constraints=labels)
|
| 131 |
elif model_name == "InfoFinder":
|
| 132 |
return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.5", device="cpu")
|
| 133 |
except Exception as e:
|