Voice_Cloner / app.py
Vageesh1's picture
Update app.py
308f38a
raw
history blame
2.69 kB
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
from elevenlabs import set_api_key
from elevenlabs import clone, generate, play
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
from pydub import AudioSegment
from huggingface_hub.hf_api import HfFolder
HfFolder.save_token('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')
access_token = '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'
import streamlit as st
#language translation function
def translate(text):
tokenizer = AutoTokenizer.from_pretrained(".",use_auth_token=True)
#model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M",use_auth_token=True)
model = AutoModelForSeq2SeqLM.from_pretrained(".",use_auth_token=True)
inputs = tokenizer(text, return_tensors="pt")
translated_tokens = model.generate(**inputs,
forced_bos_token_id=tokenizer.lang_code_to_id["hin_Deva"],
max_length=100)
result = tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]
return result
def save_uploaded_file_as_mp3(uploaded_file, output_file_path):
audio = AudioSegment.from_file(uploaded_file)
audio.export(output_file_path, format="mp3")
def ui():
st.title('Multi Lingual Voice Cloner')
st.markdown("Made by vageesh")
#audio input box
audio_file = st.file_uploader("Upload an audio file that needs to be cloned", type=[ "wav,Mp4","Mp3"])
if audio_file is not None:
output_file_path = "./output_audio.mp3"
save_uploaded_file_as_mp3(audio_file, output_file_path)
st.success(f"Audio file saved as {output_file_path}")
user_api_key = st.sidebar.text_input(
label="#### Your Eleven Labs API key here 👇",
placeholder="Paste your Eleven Labs API key API key, sk-",
type="password")
if user_api_key is not None and user_api_key.strip() != "":
set_api_key(user_api_key)
#making an voice
voice = clone(
name="Alex",
description="An middle aged American male voice with a slight hoarseness in his throat. Perfect for npodcast", # Optional
files="./output_audio.mp3",
model='eleven_multilingual_v1',
language='Hindi'
)
in_text=st.text_input("Paste the text you want to hear from english to hindi")
if in_text is not None and in_text.strip() != "":
hin_text=translate(in_text)
audio = generate(text=hin_text, voice=voice)
st.audio(audio)
if __name__=="__main__":
ui()