jmisidro commited on
Commit
af0ea77
·
verified ·
1 Parent(s): ca015ea

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -105,7 +105,7 @@ st.markdown("""
105
  @st.cache_resource(show_spinner="🔄 Loading models...")
106
  def load_models():
107
  """Load Hugging Face model and local metadata"""
108
- model_name = "liaad/Citilink-BERTimbau-large-Topic-Classification-pt"
109
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
110
 
111
  # Load HF model
 
105
  @st.cache_resource(show_spinner="🔄 Loading models...")
106
  def load_models():
107
  """Load Hugging Face model and local metadata"""
108
+ model_name = "inesctec/Citilink-BERTimbau-large-Topic-Classification-pt"
109
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
110
 
111
  # Load HF model