Add checkpoint with Git LFS

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+checkpoints/ckpt_025000_inference.pt filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -2,6 +2,3 @@ __pycache__/
 *.py[cod]
 .gradio/
 .ipynb_checkpoints/
-checkpoints/*.pt
-checkpoints/*.pth
-checkpoints/*.ckpt

app.py

@@ -142,7 +142,7 @@ def synthesize(
 with gr.Blocks(title="nanoTTS") as demo:
     gr.Markdown(
         "# nanoTTS\n"
-        "A minimal GPT-style text-to-speech model trained on LibriTTS text/audio tokens."
+        "A minimal GPT-style text-to-speech model trained on LibriTTS. Code available at [psandovalsegura/nanoTTS](https://github.com/psandovalsegura/nanoTTS)."
     )
     with gr.Row():
         with gr.Column(scale=3):

checkpoints/ckpt_025000_inference.pt

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

scripts/upload_checkpoint.py

@@ -1,46 +0,0 @@
-from __future__ import annotations
-
-import argparse
-import os
-import sys
-from pathlib import Path
-
-from huggingface_hub import HfApi
-from huggingface_hub.errors import HfHubHTTPError
-
-DEFAULT_PATH = Path("checkpoints/ckpt_025000_inference.pt")
-
-def main() -> None:
-    parser = argparse.ArgumentParser(description="Upload the nanoTTS inference checkpoint to the Space.")
-    parser.add_argument("--path", type=Path, default=DEFAULT_PATH)
-    parser.add_argument("--repo-id", default="psando/nanoTTS")
-    parser.add_argument("--path-in-repo", default=str(DEFAULT_PATH))
-    parser.add_argument("--commit-message", default="Upload nanoTTS inference checkpoint")
-    parser.add_argument("--token", default=os.environ.get("HF_TOKEN"))
-    args = parser.parse_args()
-
-    if not args.path.exists():
-        raise FileNotFoundError(f"Checkpoint not found: {args.path}")
-
-    api = HfApi()
-    try:
-        api.upload_file(
-            path_or_fileobj=str(args.path),
-            path_in_repo=args.path_in_repo,
-            repo_id=args.repo_id,
-            repo_type="space",
-            commit_message=args.commit_message,
-            token=args.token,
-        )
-    except HfHubHTTPError as exc:
-        print(
-            "Upload failed. Log in with a Hugging Face write token or set HF_TOKEN, then rerun:\n"
-            "  python scripts/upload_checkpoint.py",
-            file=sys.stderr,
-        )
-        raise exc
-    print(f"Uploaded {args.path} to {args.repo_id}:{args.path_in_repo}")
-
-
-if __name__ == "__main__":
-    main()