HMPhuoc commited on
Commit
778927e
β€’
1 Parent(s): 87bb75f

add train folder

Browse files
app.py CHANGED
@@ -11,16 +11,16 @@ import pandas as pd
11
  import plotly.express as px
12
 
13
  #Load tokenizer
14
- fp = Path(__file__).with_name('tokenizer.pkl')
15
  with open(fp,mode="rb") as f:
16
  tokenizer = pickle.load(f)
17
 
18
  #Load LSTM
19
- fp = Path(__file__).with_name('lstm_model.h5')
20
  LSTM_model = tf.keras.models.load_model(fp, compile=True)
21
 
22
  #Load GRU
23
- fp = Path(__file__).with_name('gru_model.h5')
24
  GRU_model = load_model(fp)
25
 
26
 
 
11
  import plotly.express as px
12
 
13
  #Load tokenizer
14
+ fp = Path(__file__).with_name('train/tokenizer/tokenizer.pkl')
15
  with open(fp,mode="rb") as f:
16
  tokenizer = pickle.load(f)
17
 
18
  #Load LSTM
19
+ fp = Path(__file__).with_name('train/model/lstm_model.h5')
20
  LSTM_model = tf.keras.models.load_model(fp, compile=True)
21
 
22
  #Load GRU
23
+ fp = Path(__file__).with_name('train/model/gru_model.h5')
24
  GRU_model = load_model(fp)
25
 
26
 
gru_model.h5 β†’ train/model/gru_model.h5 RENAMED
File without changes
lstm_model.h5 β†’ train/model/lstm_model.h5 RENAMED
File without changes
tokenizer.pkl β†’ train/tokenizer/tokenizer.pkl RENAMED
File without changes