Add ChatTTS download stats (#734)

This PR ensures download stats are working for
https://huggingface.co/2Noise/ChatTTS.

To do:

- [x] I assume the repo must have `library_name: chat_tts` in its
metadata for this to work. Made a PR on the repo:
https://huggingface.co/2Noise/ChatTTS/discussions/10

---------

Co-authored-by: Lucain <[email protected]>

Author: NielsRogge

packages/tasks/src/model-libraries-snippets.ts

@@ -553,6 +553,20 @@ export const voicecraft = (model: ModelData): string[] => [
 model = VoiceCraft.from_pretrained("${model.id}")`,
 ];
 
+export const chattts = (): string[] => [
+	`import ChatTTS
+import torchaudio
+
+chat = ChatTTS.Chat()
+chat.load_models(compile=False) # Set to True for better performance
+
+texts = ["PUT YOUR TEXT HERE",]
+
+wavs = chat.infer(texts, )
+
+torchaudio.save("output1.wav", torch.from_numpy(wavs[0]), 24000)`,
+];
+
 export const mlx = (model: ModelData): string[] => [
 	`pip install huggingface_hub hf_transfer
 

packages/tasks/src/model-libraries.ts

@@ -111,6 +111,14 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 			wildcard: { path: "*.npz" },
 		},
 	},
+	chat_tts: {
+		prettyLabel: "ChatTTS",
+		repoName: "ChatTTS",
+		repoUrl: "https://github.com/2noise/ChatTTS.git",
+		filter: false,
+		countDownloads: { term: { path: "asset/GPT.pt" } },
+		snippets: snippets.chattts,
+	},
 	diffusers: {
 		prettyLabel: "Diffusers",
 		repoName: "🤗/diffusers",