cloningapp / utils.py
meraj12's picture
Create utils.py
9cc018b verified
raw
history blame
577 Bytes
import torch
import torchaudio
import soundfile as sf
from pathlib import Path
def save_uploaded_audio(uploaded_file, save_path="user_audio.wav"):
with open(save_path, "wb") as f:
f.write(uploaded_file.read())
return save_path
def record_audio(file_path="recorded.wav", duration=5, sample_rate=16000):
import sounddevice as sd
from scipy.io.wavfile import write
st.info("Recording...")
audio = sd.rec(int(duration * sample_rate), samplerate=sample_rate, channels=1)
sd.wait()
write(file_path, sample_rate, audio)
return file_path