Dua Rajper commited on
Commit
b382ad4
·
verified ·
1 Parent(s): 831716f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import os
2
  import streamlit as st
3
  from groq import Groq
4
- from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
5
  from espnet2.bin.tts_inference import Text2Speech
6
  import soundfile as sf
7
  from pydub import AudioSegment
@@ -21,7 +21,7 @@ if not GROQ_API_KEY:
21
  groq_client = Groq(api_key=GROQ_API_KEY)
22
 
23
  # Load models
24
- @st.cache(allow_output_mutation=True)
25
  def load_models():
26
  # Speech-to-Text
27
  processor = AutoProcessor.from_pretrained("openai/whisper-small")
 
1
  import os
2
  import streamlit as st
3
  from groq import Groq
4
+ from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq, pipeline
5
  from espnet2.bin.tts_inference import Text2Speech
6
  import soundfile as sf
7
  from pydub import AudioSegment
 
21
  groq_client = Groq(api_key=GROQ_API_KEY)
22
 
23
  # Load models
24
+ @st.cache_resource # Use st.cache_resource for caching models
25
  def load_models():
26
  # Speech-to-Text
27
  processor = AutoProcessor.from_pretrained("openai/whisper-small")