Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +1 -1
src/streamlit_app.py
CHANGED
|
@@ -207,7 +207,7 @@ if 'df_ner' in st.session_state and not st.session_state.df_ner.empty:
|
|
| 207 |
def load_gliner_model():
|
| 208 |
"""Initializes and caches the GLiNER model for QA."""
|
| 209 |
try:
|
| 210 |
-
return GLiNER.from_pretrained("knowledgator/gliner-multitask-
|
| 211 |
except Exception as e:
|
| 212 |
st.error(f"Error loading the GLiNER model: {e}")
|
| 213 |
st.stop()
|
|
|
|
| 207 |
def load_gliner_model():
|
| 208 |
"""Initializes and caches the GLiNER model for QA."""
|
| 209 |
try:
|
| 210 |
+
return GLiNER.from_pretrained("knowledgator/gliner-multitask-v1.0", device="cpu")
|
| 211 |
except Exception as e:
|
| 212 |
st.error(f"Error loading the GLiNER model: {e}")
|
| 213 |
st.stop()
|