iisadia commited on
Commit
abcd074
·
verified ·
1 Parent(s): 2829a73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -23,8 +23,7 @@ from transformers import pipeline
23
 
24
  @st.cache_resource
25
  def load_voice_model():
26
- if 'whisper_model' not in st.session_state:
27
- st.session_state.whisper_model = pipeline("automatic-speech-recognition", model="openai/whisper-base")
28
 
29
  def process_audio(audio_bytes):
30
  waveform, sample_rate = torchaudio.load(BytesIO(audio_bytes))
 
23
 
24
  @st.cache_resource
25
  def load_voice_model():
26
+ return pipeline("automatic-speech-recognition", model="openai/whisper-tiny")
 
27
 
28
  def process_audio(audio_bytes):
29
  waveform, sample_rate = torchaudio.load(BytesIO(audio_bytes))