Spaces:

Seetha
/

IMA-pipeline-streamlit

Sleeping

Seetha commited on Jul 28, 2023

Commit

997e40f

•

1 Parent(s): 51f9b20

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -171,7 +171,7 @@ def main():
   #filename = 'Checkpoint-classification.sav'
   filename = 'model.bin'
-  count_vect = CountVectorizer(ngram_range=[1,3])
   tfidf_transformer=TfidfTransformer()
   #loaded_model = pickle.load(open(filename, 'rb'))
   #loaded_model = pickle.load(open(filename, 'rb'))

   #filename = 'Checkpoint-classification.sav'
   filename = 'model.bin'
+  count_vect = CountVectorizer(ngram_range=(1,3))
   tfidf_transformer=TfidfTransformer()
   #loaded_model = pickle.load(open(filename, 'rb'))
   #loaded_model = pickle.load(open(filename, 'rb'))