khalidey's picture
Update app.py
6f05311
raw
history blame
830 Bytes
from transformers import pipeline
import gradio as gr
pipe1 = pipeline(model="khalidey/ID2223_Lab2_Whisper_SV") # change to "your-username/the-name-you-picked"
pipe2 = pipeline('text-generation', model='gpt2')
def transcribe(audio):
text = pipe1(audio)["text"]
return text
def generateText(audio):
text = pipe1(audio)["text"]
print(text)
#generated_text = pipe2(text, max_length = 30, num_return_sequences=2)[0]['generated_text']
#print(generated_text)
generated_text='lol'
return text, generated_text
iface = gr.Interface(
fn=generateText,
inputs=gr.Audio(source="microphone", type="filepath"),
outputs=['text', 'text'],
title="Whisper Small Swedish",
description="Realtime demo for Swedish speech recognition using a fine-tuned Whisper small model.",
)
iface.launch()