Mpavan45 commited on
Commit
1929300
·
verified ·
1 Parent(s): 9d28303

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,8 @@ with open("preprocessing1.pkl", "rb") as f:
11
  clean_text = dill.load(f)
12
 
13
  # Load Text Vectorization Layer from SavedModel
14
- vectorizer = tf.saved_model.load("vector")
 
15
 
16
  # Define News Categories
17
  news_categories = ["Business", "Sci/Tech", "Sports", "World"]
 
11
  clean_text = dill.load(f)
12
 
13
  # Load Text Vectorization Layer from SavedModel
14
+ with open("vector.pkl", "rb") as f:
15
+ vectorizer = dill.load(f)
16
 
17
  # Define News Categories
18
  news_categories = ["Business", "Sci/Tech", "Sports", "World"]