Subida inicial del modelo

inference.py

@@ -0,0 +1,15 @@
+import pickle
+from sklearn.feature_extraction.text import TfidfVectorizer
+
+# Cargar el modelo y el vectorizador
+with open('vectorizer.pkl', 'rb') as f:
+    vectorizer = pickle.load(f)
+
+with open('model.pkl', 'rb') as f:
+    model = pickle.load(f)
+
+def predict(text):
+    # Preprocesar el texto usando el vectorizador
+    X = vectorizer.transform([text])
+    # Hacer una predicción usando el modelo
+    return model.predict(X)

requirements.txt

@@ -0,0 +1,5 @@
+pandas
+scikit-learn
+imbalanced-learn
+joblib
+numpy