From c44df99d499679c8e51f5ac0a7681a40ae76af50 Mon Sep 17 00:00:00 2001
From: Niels <niels.rogge1@gmail.com>
Date: Sun, 7 Apr 2024 20:34:27 +0200
Subject: [PATCH] Add tags

---
 models/voicecraft.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/models/voicecraft.py b/models/voicecraft.py
index 8ea85ad..ee8e5a7 100644
--- a/models/voicecraft.py
+++ b/models/voicecraft.py
@@ -1414,7 +1414,9 @@ class VoiceCraft(nn.Module):
         return res, flatten_gen[0].unsqueeze(0)
     
 
-class VoiceCraftHF(VoiceCraft, PyTorchModelHubMixin):
+class VoiceCraftHF(VoiceCraft, PyTorchModelHubMixin,
+                   repo_url="https://github.com/jasonppy/VoiceCraft",
+                   tags=["Text-to-Speech, VoiceCraft"]):
     def __init__(self, config: dict):
         args = Namespace(**config)
         super().__init__(args)
\ No newline at end of file