add infrastructure to automatically download the models

README.md

@@ -1,8 +1,8 @@
 ---
-title: IMS Toucan
+title: IMS Toucan Multilingual TTS
 emoji: 🦜
-colorFrom: red
-colorTo: orange
+colorFrom: green
+colorTo: yellow
 sdk: gradio
 app_file: app.py
 pinned: false

app.py

@@ -1,3 +1,6 @@
+import os
+
+import gdown
 import gradio as gr
 import numpy as np
 import torch
@@ -26,6 +29,10 @@ class TTS_Interface:
     def __init__(self):
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         self.model = Meta_FastSpeech2(device=self.device)
+        os.makedirs("Models/HiFiGAN_combined", exist_ok=True)
+        os.makedirs("Models/FastSpeech2_Meta", exist_ok=True)
+        gdown.download(url="https://drive.google.com/uc?id=1-AhjmCR6DDI6rtzPIn9ksOxQyHKf6CbG", output="Models/FastSpeech2_Meta/best.pt")
+        gdown.download(url="https://drive.google.com/uc?id=1-5sP-0JDUvKTjxhO3hUVJgArSUjuhU6P", output="Models/HiFiGAN_combined/best.pt")
 
     def read(self, prompt, language):
         language_id_lookup = {