Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -9,11 +9,11 @@ from waitress import serve
|
|
9 |
|
10 |
app = Flask(__name__)
|
11 |
|
12 |
-
# Load Whisper ASR Model for
|
13 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
14 |
-
asr_model = pipeline("automatic-speech-recognition", model="openai/whisper-
|
15 |
|
16 |
-
# Function to generate
|
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))
|