Mpavan45 commited on
Commit
ce76583
·
verified ·
1 Parent(s): b23de27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -235,7 +235,7 @@ def predict_category(text):
235
  processed_text = [pre_process(text)]
236
 
237
  # Vectorize the input
238
- text_vectorized = vectorizer.transform(processed_text).toarray()
239
 
240
  # Pad the sequence
241
  text_vectorized = pad_sequences(text_vectorized, padding='pre', maxlen=128)
 
235
  processed_text = [pre_process(text)]
236
 
237
  # Vectorize the input
238
+ text_vectorized = vectorizer(processed_text).np.tolist()
239
 
240
  # Pad the sequence
241
  text_vectorized = pad_sequences(text_vectorized, padding='pre', maxlen=128)