Update app.py

app.py

@@ -32,35 +32,44 @@ MODELS_PATH = os.getenv("MODELS_PATH", "./pretrained_models")
 os.makedirs(MODELS_PATH, exist_ok=True)
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-# download wmencodec
-url = "https://huggingface.co/westbrook/SSR-Speech-English/resolve/main/wmencodec.th"
-filename = os.path.join(MODELS_PATH, "wmencodec.th")
-response = requests.get(url, stream=True)
-response.raise_for_status()
-with open(filename, "wb") as file:
-    for chunk in response.iter_content(chunk_size=8192):
-        file.write(chunk)
-print(f"File downloaded to: {filename}")
-
-# download english model
-url = "https://huggingface.co/westbrook/SSR-Speech-English/resolve/main/English.pth"
-filename = os.path.join(MODELS_PATH, "English.th")
-response = requests.get(url, stream=True)
-response.raise_for_status()
-with open(filename, "wb") as file:
-    for chunk in response.iter_content(chunk_size=8192):
-        file.write(chunk)
-print(f"File downloaded to: {filename}")
-
-# download mandarin model
-url = "https://huggingface.co/westbrook/SSR-Speech-Mandarin/resolve/main/Mandarin.pth"
-filename = os.path.join(MODELS_PATH, "Mandarin.pth")
-response = requests.get(url, stream=True)
-response.raise_for_status()
-with open(filename, "wb") as file:
-    for chunk in response.iter_content(chunk_size=8192):
-        file.write(chunk)
-print(f"File downloaded to: {filename}")
+if not os.path.exists(os.path.join(MODELS_PATH, "wmencodec.th")):
+    # download wmencodec
+    url = "https://huggingface.co/westbrook/SSR-Speech-English/resolve/main/wmencodec.th"
+    filename = os.path.join(MODELS_PATH, "wmencodec.th")
+    response = requests.get(url, stream=True)
+    response.raise_for_status()
+    with open(filename, "wb") as file:
+        for chunk in response.iter_content(chunk_size=8192):
+            file.write(chunk)
+    print(f"File downloaded to: {filename}")
+else:
+    print("wmencodec model found")
+
+if not os.path.exists(os.path.join(MODELS_PATH, "English.pth")):
+    # download english model
+    url = "https://huggingface.co/westbrook/SSR-Speech-English/resolve/main/English.pth"
+    filename = os.path.join(MODELS_PATH, "English.pth")
+    response = requests.get(url, stream=True)
+    response.raise_for_status()
+    with open(filename, "wb") as file:
+        for chunk in response.iter_content(chunk_size=8192):
+            file.write(chunk)
+    print(f"File downloaded to: {filename}")
+else:
+    print("english model found")
+
+if not os.path.exists(os.path.join(MODELS_PATH, "Mandarin.pth")):
+    # download mandarin model
+    url = "https://huggingface.co/westbrook/SSR-Speech-Mandarin/resolve/main/Mandarin.pth"
+    filename = os.path.join(MODELS_PATH, "Mandarin.pth")
+    response = requests.get(url, stream=True)
+    response.raise_for_status()
+    with open(filename, "wb") as file:
+        for chunk in response.iter_content(chunk_size=8192):
+            file.write(chunk)
+    print(f"File downloaded to: {filename}")
+else:
+    print("mandarin model found")
 
 def get_random_string():
     return "".join(str(uuid.uuid4()).split("-"))