DrishtiSharma commited on
Commit
4c16333
1 Parent(s): 2e817f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def predict_sentiment(text):
19
  sequence_test = tokenizer.texts_to_sequences([text])
20
  padded_test = pad_sequences(sequence_test, maxlen= 52)
21
  text=padded_test
22
- model = tf.keras.models.load_model("huggingface/keras-io/bidirectional-lstm-imdb")
23
  X = [text for _ in range(len(model.input))]
24
  a=model.predict(X, verbose=0)
25
  return sentiment[np.around(a, decimals=0).argmax(axis=1)[0]]
 
19
  sequence_test = tokenizer.texts_to_sequences([text])
20
  padded_test = pad_sequences(sequence_test, maxlen= 52)
21
  text=padded_test
22
+ model = tf.keras.models.load_model("keras-io/bidirectional-lstm-imdb")
23
  X = [text for _ in range(len(model.input))]
24
  a=model.predict(X, verbose=0)
25
  return sentiment[np.around(a, decimals=0).argmax(axis=1)[0]]