Spaces:
Sleeping
Sleeping
Added the models
Browse files
app.py
CHANGED
|
@@ -9,8 +9,8 @@ def clean_text(text:str)->str:
|
|
| 9 |
text = ''.join(char for char in text if char.isalpha() or char.isspace())
|
| 10 |
return text
|
| 11 |
|
| 12 |
-
LogisticModel = joblib.load(r'
|
| 13 |
-
Vectorizer = joblib.load(r'
|
| 14 |
|
| 15 |
def predict_category(description:str)->str:
|
| 16 |
cleaned_text = clean_text(description)
|
|
|
|
| 9 |
text = ''.join(char for char in text if char.isalpha() or char.isspace())
|
| 10 |
return text
|
| 11 |
|
| 12 |
+
LogisticModel = joblib.load(r'transaction_classifier_model_Logistic_Regression.joblib')
|
| 13 |
+
Vectorizer = joblib.load(r'tfidf_vectorizer.joblib')
|
| 14 |
|
| 15 |
def predict_category(description:str)->str:
|
| 16 |
cleaned_text = clean_text(description)
|
assets/tfidf_vectorizer.joblib → tfidf_vectorizer.joblib
RENAMED
|
File without changes
|
assets/transaction_classifier_model_Logistic_Regression.joblib → transaction_classifier_model_Logistic_Regression.joblib
RENAMED
|
File without changes
|