fix

README.md

@@ -1,8 +1,8 @@
 ---
-title: Aet_demo
-emoji: 🌍
-colorFrom: gray
-colorTo: blue
+title: Selfremaster_aet_demo
+emoji: 👀
+colorFrom: blue
+colorTo: green
 sdk: gradio
 sdk_version: 2.8.12
 app_file: app.py

app.py

@@ -5,6 +5,7 @@ import torchaudio
 import numpy as np
 from lightning_module import SSLDualLightningModule
 import gradio as gr
+import subprocess
 
 def normalize_waveform(wav, sr, db=-3):
     wav, _ = torchaudio.sox_effects.apply_effects_tensor(
@@ -72,6 +73,8 @@ def transfer(audio):
     return sr, wav_transfer
 
 if __name__ == "__main__":
+    subprocess.run("./setup.sh")
+    subprocess.run(["apt-get", "install", "ffmpeg"])
     iface = gr.Interface(
         transfer,
         "audio",

setup.sh

@@ -0,0 +1,5 @@
+echo "Downloading pretrained model for audio effect transfer ..."
+curl -OL https://sarulab.sakura.ne.jp/saeki/selfremaster/pretrained/tono_aet_melspec.ckpt
+mv tono_aet_melspec.ckpt aet_sample/
+
+echo "Done!"