Update app.py
Browse files
app.py
CHANGED
|
@@ -21,12 +21,12 @@ from phoBERT import BERT_predict
|
|
| 21 |
# tokenizer = pickle.load(f)
|
| 22 |
|
| 23 |
#Load LSTM
|
| 24 |
-
fp = Path(__file__).with_name('lstm_model.keras')
|
| 25 |
-
LSTM_model = tf.keras.models.load_model(
|
| 26 |
|
| 27 |
#Load GRU
|
| 28 |
-
fp = Path(__file__).with_name('gru_model.keras')
|
| 29 |
-
GRU_model = tf.keras.models.load_model(
|
| 30 |
|
| 31 |
|
| 32 |
def tokenizer_pad(tokenizer,comment_text,max_length=200):
|
|
|
|
| 21 |
# tokenizer = pickle.load(f)
|
| 22 |
|
| 23 |
#Load LSTM
|
| 24 |
+
# fp = Path(__file__).with_name('lstm_model.keras')
|
| 25 |
+
LSTM_model = tf.keras.models.load_model('lstm_model.keras')
|
| 26 |
|
| 27 |
#Load GRU
|
| 28 |
+
# fp = Path(__file__).with_name('gru_model.keras')
|
| 29 |
+
GRU_model = tf.keras.models.load_model('gru_model.keras')
|
| 30 |
|
| 31 |
|
| 32 |
def tokenizer_pad(tokenizer,comment_text,max_length=200):
|