Hugging Face's logo

Spaces:
arjahojnik
/
RNN-sentiment-demo
Sleeping

App Files Community
arjahojnik commited on
Commit
bac4aa5
·
verified ·
1 Parent(s): a2b5272

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -3,11 +3,14 @@ import numpy as np
3
  from tensorflow.keras.models import load_model
4
  from tensorflow.keras.preprocessing.sequence import pad_sequences
5
  import pickle
 
 
 
6
 
7
 
8
  gru_model = load_model("best_GRU_tuning_model.h5")
9
- lstm_model = load_model("lstm_model.h5")
10
- bilstm_model = load_model("bilstm_model.h5")
11
 
12
 
13
  with open("my_tokenizer.pkl", "rb") as f:
 
3
  from tensorflow.keras.models import load_model
4
  from tensorflow.keras.preprocessing.sequence import pad_sequences
5
  import pickle
6
+ import os
7
+
8
+ os.environ["CUDA_VISIBLE_DEVICES"] = "-1" # Disable GPU
9
 
10
 
11
  gru_model = load_model("best_GRU_tuning_model.h5")
12
+ lstm_model = load_model("LSTM_model.h5")
13
+ bilstm_model = load_model("BiLSTM_model.h5")
14
 
15
 
16
  with open("my_tokenizer.pkl", "rb") as f: