shangeth commited on
Commit
517b9a4
·
verified ·
1 Parent(s): e94759d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -25,7 +25,8 @@ def plot_mel_spectrogram(mel_spec):
25
  def get_or_load_model():
26
  if 'model' not in st.session_state or 'tokenizer' not in st.session_state or 'processor' not in st.session_state:
27
  ckpt_path = "checkpoints/pretrained_checkpoint.ckpt"
28
- model = SpeechLLMLightning.load_from_checkpoint(ckpt_path)
 
29
  tokenizer = model.llm_tokenizer
30
  model.eval()
31
  model.freeze()
 
25
  def get_or_load_model():
26
  if 'model' not in st.session_state or 'tokenizer' not in st.session_state or 'processor' not in st.session_state:
27
  ckpt_path = "checkpoints/pretrained_checkpoint.ckpt"
28
+ model = SpeechLLMLightning()
29
+ # .load_from_checkpoint(ckpt_path)
30
  tokenizer = model.llm_tokenizer
31
  model.eval()
32
  model.freeze()