fix

app.py

@@ -6,6 +6,7 @@ import numpy as np
 from lightning_module import SSLDualLightningModule
 import gradio as gr
 import subprocess
+import requests
 
 def normalize_waveform(wav, sr, db=-3):
     wav, _ = torchaudio.sox_effects.apply_effects_tensor(
@@ -15,6 +16,35 @@ def normalize_waveform(wav, sr, db=-3):
     )
     return wav.squeeze(0)
 
+def download_file_from_google_drive(id, destination):
+    URL = "https://docs.google.com/uc?export=download"
+
+    session = requests.Session()
+
+    response = session.get(URL, params = { 'id' : id }, stream = True)
+    token = get_confirm_token(response)
+
+    if token:
+        params = { 'id' : id, 'confirm' : token }
+        response = session.get(URL, params = params, stream = True)
+
+    save_response_content(response, destination)    
+
+def get_confirm_token(response):
+    for key, value in response.cookies.items():
+        if key.startswith('download_warning'):
+            return value
+
+    return None
+
+def save_response_content(response, destination):
+    CHUNK_SIZE = 32768
+
+    with open(destination, "wb") as f:
+        for chunk in response.iter_content(CHUNK_SIZE):
+            if chunk: # filter out keep-alive new chunks
+                f.write(chunk)
+
 def calc_spectrogram(wav, config):
     spec_module = torchaudio.transforms.MelSpectrogram(
             sample_rate=config["preprocess"]["sampling_rate"],
@@ -74,6 +104,7 @@ def transfer(audio):
 
 if __name__ == "__main__":
     subprocess.run(["curl", "-OL", "https://sarulab.sakura.ne.jp/saeki/selfremaster/pretrained/tono_aet_melspec.ckpt"])
+    download_file_from_google_drive("10OJ2iznutxzp8MEIS6lBVaIS_g5c_70V", "hifigan/hifigan_melspec_universal")
     subprocess.run(["ls", "-a"])
     subprocess.run(["ls", "-a", "aet_sample"])