DrishtiSharma commited on
Commit
5886659
1 Parent(s): 6b08e4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -3,6 +3,8 @@ import tensorflow as tf
3
  from tensorflow.keras.preprocessing.sequence import pad_sequences
4
  import pickle
5
  from huggingface_hub import from_pretrained_keras
 
 
6
 
7
  model = from_pretrained_keras("keras-io/bidirectional-lstm-imdb")
8
 
 
3
  from tensorflow.keras.preprocessing.sequence import pad_sequences
4
  import pickle
5
  from huggingface_hub import from_pretrained_keras
6
+ import os
7
+ os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
8
 
9
  model = from_pretrained_keras("keras-io/bidirectional-lstm-imdb")
10