Download inference.py from delineiro/soflexpoc: direct link, hf CLI and curl.
- Browser
- Download file 535 Bytes
-
https://huggingface.co/delineiro/soflexpoc/resolve/main/inference.py
- Command line
-
hf download hf://delineiro/soflexpoc/inference.py
-
curl -L -o inference.py https://huggingface.co/delineiro/soflexpoc/resolve/main/inference.py
535 Bytes
| import joblib | |
| def load_model_and_vectorizer(): | |
| model = joblib.load('model.pkl') | |
| vectorizer = joblib.load('vectorizer.pkl') | |
| return model, vectorizer | |
| def predict(text): | |
| model, vectorizer = load_model_and_vectorizer() | |
| text_vectorized = vectorizer.transform([text]) | |
| prediction = model.predict(text_vectorized) | |
| return prediction[0] | |
| if __name__ == '__main__': | |
| # Example usage | |
| text = "Ejemplo de declaraci锟絥" | |
| result = predict(text) | |
| print(f"Categor锟絘 predicha: {result}") | |