azizbarank commited on
Commit
8c04409
·
1 Parent(s): 1c1adf4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -67,13 +67,13 @@ with model_results:
67
 
68
  # instantiating count vectorizor
69
  tfidf = TfidfVectorizer(stop_words=stop_words)
70
- X_train = joblib.load(open('main/X_train.pickel', 'rb'))
71
  X_test = lemmatized_output
72
  X_train_count = tfidf.fit_transform(X_train)
73
  X_test_count = tfidf.transform(X_test)
74
 
75
  # loading in model
76
- final_model = joblib.load(open('final_bayes.pickle', 'rb'))
77
 
78
  # apply model to make predictions
79
  prediction = final_model.predict(X_test_count[0])
 
67
 
68
  # instantiating count vectorizor
69
  tfidf = TfidfVectorizer(stop_words=stop_words)
70
+ X_train = joblib.load(open('resources/X_train.pickel', 'rb'))
71
  X_test = lemmatized_output
72
  X_train_count = tfidf.fit_transform(X_train)
73
  X_test_count = tfidf.transform(X_test)
74
 
75
  # loading in model
76
+ final_model = joblib.load(open('resources/final_bayes.pickle', 'rb'))
77
 
78
  # apply model to make predictions
79
  prediction = final_model.predict(X_test_count[0])