MusIre commited on
Commit
630d7c9
·
1 Parent(s): 84d561c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -9,10 +9,13 @@ subprocess.run(["pip", "install", "torch", "torchvision", "torchaudio", "-f", "h
9
 
10
  import gradio as gr
11
  import torchaudio
12
- from transformers import pipeline
 
13
 
14
  # Load the Whispy/Whisper Italian ASR model
15
- whisper_italian_asr = pipeline("whisper-italian")
 
 
16
 
17
  # Define the ASR function
18
  def transcribe_audio(audio):
 
9
 
10
  import gradio as gr
11
  import torchaudio
12
+
13
+ from transformers import AutoModelForSpeechRecognition, AutoTokenizer, pipeline
14
 
15
  # Load the Whispy/Whisper Italian ASR model
16
+ model_name = "facebook/whisper-large-italian"
17
+ whisper_italian_asr = pipeline("automatic-speech-recognition", model=model_name, device=0)
18
+
19
 
20
  # Define the ASR function
21
  def transcribe_audio(audio):