Bartusito commited on
Commit
102b45f
1 Parent(s): b175f5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,13 +1,14 @@
1
  import gradio as gr
2
  from huggingsound import SpeechRecognitionModel
3
  from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
 
4
 
5
  # Función para convertir la tasa de muestreo del audio de entrada
6
  def modelo1(audio):
7
  # Convertir la tasa de muestreo del audio
8
- model = SpeechRecognitionModel("jonatasgrosman/wav2vec2-large-xlsr-53-english")
9
- transcriptions = model.transcribe(audio)
10
- return transcriptions
11
 
12
  def modelo2(text):
13
  model_id = "stabilityai/stable-diffusion-2-1"
 
1
  import gradio as gr
2
  from huggingsound import SpeechRecognitionModel
3
  from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
4
+ from transformers import pipeline
5
 
6
  # Función para convertir la tasa de muestreo del audio de entrada
7
  def modelo1(audio):
8
  # Convertir la tasa de muestreo del audio
9
+ whisper = pipeline('automatic-speech-recognition', model = 'openai/whisper-medium', device = 0)
10
+ text = whisper(audio)
11
+ return text
12
 
13
  def modelo2(text):
14
  model_id = "stabilityai/stable-diffusion-2-1"