lokesh341 commited on
Commit
8dd6078
·
verified ·
1 Parent(s): e290660

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -9,11 +9,11 @@ from waitress import serve
9
 
10
  app = Flask(__name__)
11
 
12
- # Load Whisper ASR Model for Highly Accurate Speech-to-Text
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
- asr_model = pipeline("automatic-speech-recognition", model="openai/whisper-large-v3", device=0 if device == "cuda" else -1)
15
 
16
- # Function to generate audio prompts
17
  def generate_audio_prompt(text, filename):
18
  tts = gTTS(text=text, lang="en")
19
  tts.save(os.path.join("static", filename))
 
9
 
10
  app = Flask(__name__)
11
 
12
+ # Load Whisper ASR Model for Faster Response (Switch to medium for better speed)
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
+ asr_model = pipeline("automatic-speech-recognition", model="openai/whisper-medium", device=0 if device == "cuda" else -1)
15
 
16
+ # Function to generate voice prompts
17
  def generate_audio_prompt(text, filename):
18
  tts = gTTS(text=text, lang="en")
19
  tts.save(os.path.join("static", filename))