Tirath5504 commited on
Commit
ce881ac
·
verified ·
1 Parent(s): 972cb70

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -10,14 +10,17 @@ from tensorflow import keras
10
  from keras import layers
11
  from tensorflow.keras.preprocessing.text import Tokenizer
12
  from tensorflow.keras.preprocessing.sequence import pad_sequences
13
- import joblib
 
14
 
15
  nltk.download('stopwords')
16
  nltk.download('omw-1.4')
17
  nltk.download('wordnet')
18
  nltk.download('punkt')
19
 
20
- tokenizer, model = joblib.load("lstm_model.pkl")
 
 
21
 
22
  def preprocess(text, tokenizer):
23
  lemmatizer = WordNetLemmatizer()
 
10
  from keras import layers
11
  from tensorflow.keras.preprocessing.text import Tokenizer
12
  from tensorflow.keras.preprocessing.sequence import pad_sequences
13
+ from tensorflow.keras.models import load_model
14
+ from joblib import load
15
 
16
  nltk.download('stopwords')
17
  nltk.download('omw-1.4')
18
  nltk.download('wordnet')
19
  nltk.download('punkt')
20
 
21
+
22
+ model = load_model('model.keras')
23
+ tokenizer = load('tokenizer.pkl')
24
 
25
  def preprocess(text, tokenizer):
26
  lemmatizer = WordNetLemmatizer()