Hugging Face's logo

Spaces:
Gillie2004
/
SMS_Spam_Detection_using_SVM
Build error

App Files Community
3v324v23 commited on
Commit
eabb121
·
1 Parent(s): 1e658eb

app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -70,8 +70,8 @@ def train_and_save_model():
70
  y_pred = svm_model.predict(X_test_tfidf)
71
  accuracy = accuracy_score(y_test, y_pred)
72
 
73
- joblib.dump(svm_model, "D:/CCS229 - Intelligent System/SMS_Spam_Detection_using_SVM/svm_sms_spam.pkl")
74
- joblib.dump(vectorizer, "D:/CCS229 - Intelligent System/SMS_Spam_Detection_using_SVM/vectorizer.pkl")
75
 
76
  return svm_model, vectorizer, accuracy
77
 
 
70
  y_pred = svm_model.predict(X_test_tfidf)
71
  accuracy = accuracy_score(y_test, y_pred)
72
 
73
+ joblib.dump(svm_model, "svm_sms_spam.pkl")
74
+ joblib.dump(vectorizer, "vectorizer.pkl")
75
 
76
  return svm_model, vectorizer, accuracy
77