Update app.py

app.py

@@ -6,7 +6,7 @@ import pandas as pd
 import numpy as np
 import tensorflow as tf 
 from tensorflow.keras.preprocessing.sequence import pad_sequences
-
+from huggingface_hub.keras_mixin import from_pretrained_keras
 
 #Loading the tokenizer
 with open('tokenizer.pickle', 'rb') as f:
@@ -19,7 +19,7 @@ def predict_sentiment(text):
     sequence_test = tokenizer.texts_to_sequences([text])
     padded_test = pad_sequences(sequence_test, maxlen= 52)
     text=padded_test
-    model = tf.keras.models.load_model("keras-io/bidirectional-lstm-imdb")
+    model = from_pretrained_keras("keras-io/bidirectional-lstm-imdb")
     X = [text for _ in range(len(model.input))]
     a=model.predict(X, verbose=0)
     return sentiment[np.around(a, decimals=0).argmax(axis=1)[0]]