Update src/streamlit_app.py
Browse files- 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 = "
|
| 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
|