azizbarank commited on
Commit
7e0f5c4
1 Parent(s): e724606

Update app.py

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