eradhea commited on
Commit
2fc1304
1 Parent(s): c9016fc

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +36 -0
app.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration, pipeline
2
+ import gradio as gr
3
+ from textblob import TextBlob as tb
4
+
5
+ mname = "facebook/blenderbot-400M-distill"
6
+ model = BlenderbotForConditionalGeneration.from_pretrained(mname)
7
+ tokenizer = BlenderbotTokenizer.from_pretrained(mname)
8
+ p = pipeline("automatic-speech-recognition", model = "facebook/wav2vec2-large-xlsr-53-spanish")
9
+
10
+
11
+ def transcribe(audio): # Desde un mensaje de voz en español
12
+
13
+ mensaje_voz = p(audio)["text"]
14
+ blob = tb(mensaje_voz)
15
+ MESSAGE = str(blob.translate(from_lang='es', to='en')) # Traduce el mensaje al inglés
16
+
17
+ inputs = tokenizer(MESSAGE, return_tensors="pt") # Tokeniza el mensaje traducido
18
+ reply_ids = model.generate(**inputs)
19
+
20
+ response = tokenizer.batch_decode(reply_ids)[0] # Genera la respuesta (en inglés)
21
+
22
+ blob_2 = tb(response)
23
+ respuesta = str(blob_2.translate(from_lang='en', to='es'))[4:-5] # Traduce la respuesta al español
24
+
25
+ return respuesta
26
+
27
+
28
+ gr.Interface(
29
+ fn=transcribe,
30
+ inputs=[
31
+ gr.Audio(source="microphone", type="filepath")
32
+ ],
33
+ outputs=[
34
+ "textbox",
35
+ ],
36
+ live=True,).launch(share=True)