RwandaNameGenderModel / inference.py
benax-rw's picture
Upload folder using huggingface_hub
b805416 verified
raw
history blame contribute delete
829 Bytes
import joblib
# Load the trained model and vectorizer
model = joblib.load("model/logistic_model.joblib")
vectorizer = joblib.load("model/vectorizer.joblib")
# Function to make predictions on user input
def predict_gender(name):
name = name.strip().lower()
if not name:
return "Invalid input"
name_vector = vectorizer.transform([name])
gender_pred = model.predict(name_vector)
return gender_pred[0].capitalize()
# Interactive inference
print("RwandaNameGenderModel - Gender Prediction from Rwandan Name(s)")
while True:
user_input = input("Enter a name to predict gender (or type 'exit' to stop): ")
if user_input.lower().strip() == 'exit':
print("Exiting...")
break
gender = predict_gender(user_input)
print(f"The predicted gender for '{user_input}' is: {gender}\n")