Hugging Face's logo

Spaces:
Fluospark128
/
Mind_Aid_Demo
Sleeping

App Files Community
Fluospark128 commited on
Commit
7a1a9b5
·
verified ·
1 Parent(s): d3e5661

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import os
8
 
9
  # 1. Speech-to-Text pipeline
10
  print("Loading ASR model...")
11
- speech_to_text_pipeline = pipeline("automatic-speech-recognition", model="openai/whisper-small") #model="facebook/wav2vec2-base-960h"
12
 
13
  # 2. Text generation model (GPT2)
14
  print("Loading GPT-2 model...")
 
8
 
9
  # 1. Speech-to-Text pipeline
10
  print("Loading ASR model...")
11
+ speech_to_text_pipeline = pipeline("automatic-speech-recognition", model="facebook/wav2vec2-base-960h") #model="openai/whisper-small") #model="facebook/wav2vec2-base-960h"
12
 
13
  # 2. Text generation model (GPT2)
14
  print("Loading GPT-2 model...")