inference.py

import joblib
import numpy as np

# Cargar el vectorizador
try:
    with open('vectorizer.pkl', 'rb') as f:
        vectorizer = joblib.load(f)
except Exception as e:
    print(f"Error al cargar el vectorizador: {e}")

# Cargar el modelo
try:
    with open('model.pkl', 'rb') as f:
        model = joblib.load(f)
except Exception as e:
    print(f"Error al cargar el modelo: {e}")

def preprocess(text):
    """Preprocesa el texto para la inferencia."""
    try:
        return vectorizer.transform([text])
    except Exception as e:
        print(f"Error en el preprocesamiento: {e}")

def predict(text):
    """Realiza la predicción a partir del texto ingresado."""
    try:
        X = preprocess(text)
        return model.predict(X)
    except Exception as e:
        print(f"Error en la predicción: {e}")

if __name__ == "__main__":
    # Prueba del modelo
    test_text = "Texto de prueba para predecir"
    result = predict(test_text)
    print(f"Predicción: {result}")