import joblib | |
# Load the trained model | |
model = joblib.load('disaster_classification_model.joblib') | |
# Sample text to test | |
text = "This is a test message to check the model." | |
# Make prediction | |
prediction = model.predict([text])[0] | |
# Print result | |
result = "disaster" if prediction == 1 else "not" | |
print(f"Prediction: {result}") | |