predict_ar.py

def predict(text):
    print("text -> ",text)
    X=vectorizer.transform([text_s])
    X_pred=lp_classifier.predict(X).toarray()
    return [data.columns[i] for i,x in enumerate(X_pred[0]) if x==1]

text="كان العمال في الورشة لطفاء وكان عملهن جيد جداً"

predict(test_txt)