Text-to-Audio / app.py
MaxGab's picture
Update app.py
c7bc873 verified
raw
history blame
349 Bytes
import gradio as gr
from transformers import pipeline
if tokenizer.pad_token is None:
tokenizer.pad_token = tokenizer.eos_token
model.config.pad_token_id = model.config.eos_token_id
pipe = pipeline(model="suno/bark-small")
output = pipe("Hey it's HuggingFace on the phone!")
audio = output["audio"]
sampling_rate = output["sampling_rate"]