Spaces:

Yash008
/

comment-toxicity-detection

Sleeping

Yash008 commited on Sep 26, 2024

Commit

5f7b2d6

verified ·

1 Parent(s): d6c5ac9

Update helper.py

Files changed (1) hide show

helper.py CHANGED Viewed

@@ -10,7 +10,7 @@ nltk.download('stopwords')
 stop_words = set(stopwords.words('english'))
 def load_vectorizer():
-    from_disk = pickle.load(open("models/tv_layer.pkl", "rb"))
     new_v = TextVectorization.from_config(from_disk['config'])
     # You have to call `adapt` with some dummy data (BUG in Keras)
     new_v.adapt(tf.data.Dataset.from_tensor_slices(["xyz"]))

 stop_words = set(stopwords.words('english'))
 def load_vectorizer():
+    from_disk = pickle.load(open("tv_layer.pkl", "rb"))
     new_v = TextVectorization.from_config(from_disk['config'])
     # You have to call `adapt` with some dummy data (BUG in Keras)
     new_v.adapt(tf.data.Dataset.from_tensor_slices(["xyz"]))