Spaces:

Nacho86
/

alertDetection

Sleeping

alertDetection / secciones /model_utils.py

nacho

vfs

7bf61c5 almost 3 years ago

274 Bytes

import joblib
import streamlit as st


# Cargar el modelo y el vectorizador
@st.cache_data
def cargar_modelo_y_vectorizador():
    modelo = joblib.load('modelo_nn_g2.pkl')
    tfidf_vectorizador = joblib.load('tfidf_vectorizador2.pkl')
    return modelo, tfidf_vectorizador

	import joblib
	import streamlit as st


	# Cargar el modelo y el vectorizador
	@st.cache_data
	def cargar_modelo_y_vectorizador():
	modelo = joblib.load('modelo_nn_g2.pkl')
	tfidf_vectorizador = joblib.load('tfidf_vectorizador2.pkl')
	return modelo, tfidf_vectorizador