Update app.py

app.py

@@ -5,17 +5,13 @@ import torch
 import uuid
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-
 try:
     model = ChatterboxTTS.from_pretrained(device=device)
 except:
     pass
 
 def generate_audio(text, lang_code):
-    if not lang_code:
-        lang_code = "en"
-    
-    wav = model.generate(text, language=lang_code)
+    wav = model.generate(text)
     
     filename = f"/tmp/{uuid.uuid4()}.wav"
     torchaudio.save(filename, wav, model.sr)
@@ -25,8 +21,8 @@ def generate_audio(text, lang_code):
 iface = gr.Interface(
     fn=generate_audio,
     inputs=[
-        gr.Textbox(lines=3),
-        gr.Dropdown(choices=["ar", "en"], value="ar")
+        gr.Textbox(lines=3, label="Text"),
+        gr.Dropdown(choices=["ar", "en"], value="ar", label="Language (Just for reference)")
     ],
     outputs=gr.Audio(type="filepath"),
     allow_flagging="never"