harpalsinh8 commited on
Commit
6998eea
·
verified ·
1 Parent(s): d762f22

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -0
app.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import torch
3
+ from transformers import AutoModelForTextToWaveform, AutoTokenizer
4
+
5
+ model_id = "kyutai/tts-1.6b-en_fr"
6
+ model = AutoModelForTextToWaveform.from_pretrained(model_id)
7
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
8
+
9
+ def tts_fn(text, language="en"):
10
+ inputs = tokenizer(text, return_tensors="pt")
11
+ with torch.no_grad():
12
+ audio = model.generate(**inputs, language=language)
13
+ return (16000, audio.cpu().numpy())
14
+
15
+ demo = gr.Interface(
16
+ fn=tts_fn,
17
+ inputs=[
18
+ gr.Textbox(label="Enter text"),
19
+ gr.Dropdown(["en", "fr"], value="en", label="Language"),
20
+ ],
21
+ outputs=gr.Audio(label="Generated Speech"),
22
+ title="Kyutai TTS Free Demo (CPU)",
23
+ description="⚠️ Running on free CPU → slow, but works!"
24
+ )
25
+
26
+ if __name__ == "__main__":
27
+ demo.launch()