Hugging Face's logo

Spaces:
EasonYan
/
MTECBS
Sleeping

App Files Community
yan123yan commited on
Commit
dfa4f89
·
1 Parent(s): 6e8e2df

fix bugs

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ with st.sidebar:
61
  ## LSTM CPU Inference
62
  #################################################
63
  lstm_cpu_ckpt_file = os.path.join("model", "lstm.ckpt")
64
- lstm_cpu_model = LSTMModel.load_from_checkpoint(lstm_ckpt_file)
65
  lstm_cpu_model.to("cpu")
66
  lstm_cpu_model.eval()
67
  lstm_cpu_start_time = time.time()
 
61
  ## LSTM CPU Inference
62
  #################################################
63
  lstm_cpu_ckpt_file = os.path.join("model", "lstm.ckpt")
64
+ lstm_cpu_model = LSTMModel.load_from_checkpoint(lstm_cpu_ckpt_file)
65
  lstm_cpu_model.to("cpu")
66
  lstm_cpu_model.eval()
67
  lstm_cpu_start_time = time.time()