Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,4 +1,8 @@
 import os
+os.environ['HF_HOME'] = '/tmp'
+
+
+
 import time
 import streamlit as st
 import pandas as pd
@@ -12,8 +16,7 @@ from gliner import GLiNER
 from comet_ml import Experiment
 import hashlib
 
-# Set up environment variables
-os.environ['HF_HOME'] = '/tmp'
+
 
 # --- Page Configuration and UI Elements ---
 st.set_page_config(layout="wide", page_title="Named Entity Recognition App")
@@ -207,7 +210,7 @@ if 'df_ner' in st.session_state and not st.session_state.df_ner.empty:
 def load_gliner_model():
     """Initializes and caches the GLiNER model for QA."""
     try:
-        return GLiNER.from_pretrained("knowledgator/gliner-multitask-v1.0", device="cpu")
+        return GLiNER.from_pretrained("knowledgator/gliner-multitask-v2.0", device="cpu")
     except Exception as e:
         st.error(f"Error loading the GLiNER model: {e}")
         st.stop()