rahgadda commited on
Commit
63bb482
·
verified ·
1 Parent(s): eb9e37c

Initial Draft

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -1,14 +1,15 @@
 
1
  import streamlit as st
2
  import speech_recognition as sr
3
  from streaming_stt_nemo import Model
4
 
5
- if "sv_load_flash-attn" not in st.session_state:
6
- subprocess.run(
7
- "pip install flash-attn --no-build-isolation",
8
- env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
9
- shell=True,
10
- )
11
- st.session_state.sv_load_flash-attn = True
12
 
13
  def main():
14
  st.title("Speech to Text Converter")
 
1
+ import subprocess
2
  import streamlit as st
3
  import speech_recognition as sr
4
  from streaming_stt_nemo import Model
5
 
6
+ if "sv_load_flash_attention" not in st.session_state:
7
+ subprocess.run(
8
+ "pip install flash-attn --no-build-isolation",
9
+ env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
10
+ shell=True,
11
+ )
12
+ st.session_state.sv_load_flash_attention = True
13
 
14
  def main():
15
  st.title("Speech to Text Converter")