Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -21,11 +21,11 @@ from underthesea import word_tokenize
|
|
21 |
# tokenizer = pickle.load(f)
|
22 |
|
23 |
#Load LSTM
|
24 |
-
fp = Path(__file__).with_name('lstm_model.
|
25 |
LSTM_model = tf.keras.models.load_model(fp)
|
26 |
|
27 |
#Load GRU
|
28 |
-
fp = Path(__file__).with_name('gru_model.
|
29 |
GRU_model = tf.keras.models.load_model(fp)
|
30 |
|
31 |
|
|
|
21 |
# tokenizer = pickle.load(f)
|
22 |
|
23 |
#Load LSTM
|
24 |
+
fp = Path(__file__).with_name('lstm_model.h5')
|
25 |
LSTM_model = tf.keras.models.load_model(fp)
|
26 |
|
27 |
#Load GRU
|
28 |
+
fp = Path(__file__).with_name('gru_model.h5')
|
29 |
GRU_model = tf.keras.models.load_model(fp)
|
30 |
|
31 |
|