Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +1 -1
src/streamlit_app.py
CHANGED
|
@@ -107,7 +107,7 @@ def load_gliner_model(model_name):
|
|
| 107 |
if model_name == "HR_AI":
|
| 108 |
return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.5", nested_ner=True, num_gen_sequences=2, gen_constraints=labels)
|
| 109 |
elif model_name == "InfoFinder":
|
| 110 |
-
return GLiNER.from_pretrained("knowledgator/gliner-multitask-
|
| 111 |
except Exception as e:
|
| 112 |
st.error(f"Error loading the GLiNER model: {e}")
|
| 113 |
st.stop()
|
|
|
|
| 107 |
if model_name == "HR_AI":
|
| 108 |
return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.5", nested_ner=True, num_gen_sequences=2, gen_constraints=labels)
|
| 109 |
elif model_name == "InfoFinder":
|
| 110 |
+
return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.5", device="cpu")
|
| 111 |
except Exception as e:
|
| 112 |
st.error(f"Error loading the GLiNER model: {e}")
|
| 113 |
st.stop()
|