Update

.gitignore

@@ -1,5 +1,4 @@
 gradio_cached_examples/
-models
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

Dockerfile

@@ -45,9 +45,8 @@ RUN pyenv install ${PYTHON_VERSION} && \
 
 COPY --chown=1000 . ${HOME}/app
 RUN pip install -r ${HOME}/app/requirements.txt && \
-    pip install ${HOME}/app/wheels/fairseq2n-0.2.0.dev0-cp310-cp310-linux_x86_64.whl && \
-    pip install ${HOME}/app/wheels/fairseq2-0.2.0.dev0-py3-none-any.whl && \
-    pip install ${HOME}/app/wheels/seamless_communication-1.0.0-py3-none-any.whl
+    pip install fairseq2 --pre --extra-index-url https://fair.pkg.atmeta.com/fairseq2/whl/nightly/pt2.1.0/cu121 && \
+    pip install ${HOME}/app/whl/seamless_communication-1.0.0-py3-none-any.whl
 
 ENV PYTHONPATH=${HOME}/app \
     PYTHONUNBUFFERED=1 \

app.py

@@ -7,6 +7,7 @@ import gradio as gr
 import numpy as np
 import torch
 import torchaudio
+from fairseq2.assets import InProcAssetMetadataProvider, asset_store
 from huggingface_hub import snapshot_download
 from seamless_communication.inference import Translator
 
@@ -20,8 +21,23 @@ from lang_list import (
     TEXT_SOURCE_LANGUAGE_NAMES,
 )
 
-if not pathlib.Path("models").exists():
-    snapshot_download(repo_id="meta-private/M4Tv2", repo_type="model", local_dir="models")
+CHECKPOINTS_PATH = pathlib.Path(os.getenv("CHECKPOINTS_PATH", "/home/user/app/models"))
+if not CHECKPOINTS_PATH.exists():
+    snapshot_download(repo_id="meta-private/M4Tv2", repo_type="model", local_dir=CHECKPOINTS_PATH)
+asset_store.env_resolvers.clear()
+asset_store.env_resolvers.append(lambda: "demo")
+demo_metadata = [
+    {
+        "name": "seamlessM4T_v2_large@demo",
+        "checkpoint": f"file://{CHECKPOINTS_PATH}/seamlessM4T_v2_large.pt",
+        "char_tokenizer": f"file://{CHECKPOINTS_PATH}/spm_char_lang38_tc.model",
+    },
+    {
+        "name": "vocoder_v2@demo",
+        "checkpoint": f"file://{CHECKPOINTS_PATH}/vocoder_v2.pt",
+    },
+]
+asset_store.metadata_providers.append(InProcAssetMetadataProvider(demo_metadata))
 
 DESCRIPTION = """\
 # SeamlessM4T

wheels/fairseq2-0.2.0.dev0-py3-none-any.whl

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:0ef5a6eef679333aeb08d24bee6a054120bd7d29d67a60c4a73ca4aaa5bcc261
-size 187508

wheels/fairseq2n-0.2.0.dev0-cp310-cp310-linux_x86_64.whl

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:4865d75ce2a48aaf7b7416c7b109dec123fdc577186b9cabe2d83b74cb27d3fa
-size 2213239

{wheels → whl}/seamless_communication-1.0.0-py3-none-any.whl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3a6641670c404c2eac0c2893f498fc350ca22c5d4c9c67afcec18a578563cb3f
-size 121560
+oid sha256:c3380dc7d6613c4dc9ef4d78fd3dcf1a50f7c6a659b8ba0b37ad5237533d002e
+size 234811