Update app.py

app.py

@@ -2,16 +2,21 @@ import torch
 import numpy as np
 import gradio as gr
 from transformers import pipeline
+import logging
+
+# -----------------------------
+# Reduce Transformers warnings
+# -----------------------------
+logging.getLogger("transformers").setLevel(logging.ERROR)
 
 # -----------------------------
 # LOAD PIPELINE (HF AUTH REQUIRED)
 # -----------------------------
 device = 0 if torch.cuda.is_available() else -1
-
 tts_pipe = pipeline(
     task="text-to-speech",
     model="canopylabs/orpheus-3b-0.1-ft",
-    device=device,
+    device=device
 )
 
 # -----------------------------
@@ -21,7 +26,11 @@ def tts_generate(text):
     if not text.strip():
         return None
 
-    output = tts_pipe(text)
+    try:
+        output = tts_pipe(text)
+    except Exception as e:
+        print("Error:", e)
+        return None
 
     audio = np.asarray(output["audio"], dtype=np.float32)
     sr = output["sampling_rate"]
@@ -29,45 +38,44 @@ def tts_generate(text):
     return (sr, audio)
 
 # -----------------------------
-# SAMPLE TEXTS WITH TAGS
+# SAMPLE TEXTS WITH EXPRESSIVE TAGS
 # -----------------------------
 SAMPLES = [
-    "Just end up crashing somewhere. <laughs> No, because remember last time? You fell asleep—",
-    "But now that the cat's out of the bag, we can be the couple that we were always destined to be.",
-    "Running through the grass, playing under the falling leaves. <laughs> My sweet little kit, the—",
-    "Deal with it. I will. I'll just scowl and watch TV by myself <sighs>.",
-    "Hmm… I don't know. <nervous laughter> This feels like a bad idea.",
-    "I'm so tired today <yawning> but I still have so much work to do.",
-    "Wait—did you hear that? <gasps> I swear something just moved.",
-    "<whispers> Don't turn around. Just keep walking.",
-    "Ugh… <scoffs> I can't believe this is happening again.",
-    "Okay okay <laughs nervously> maybe it wasn't my best decision."
+    "Just end up crashing somewhere. <laugh> No, because remember last time? You fell asleep—",
+    "Running through the grass, playing under the falling leaves. <chuckle> My sweet little kit, the—",
+    "Deal with it. I will. I'll just scowl and watch TV by myself <sigh>.",
+    "Hmm… I don't know. <laugh> This feels like a bad idea. <gasp>",
+    "I'm so tired today <yawn> but I still have so much work to do.",
+    "<cough> Wait—did you hear that? <gasp> I swear something just moved.",
+    "<whisper> Don't turn around. Just keep walking.",
+    "Ugh… <sigh> I can't believe this is happening again.",
+    "Okay okay <laughs nervously> maybe it wasn't my best decision.",
+    "I just got a new sword, ooh, it's a mighty Anduril! <groan>"
 ]
 
 # -----------------------------
-# GRADIO UI
+# GRADIO INTERFACE
 # -----------------------------
 demo = gr.Interface(
     fn=tts_generate,
     inputs=gr.Textbox(
-        label="Enter text (use expressive tags like <laughs>, <sighs>)",
+        label="Enter text (use expressive tags like <laugh>, <sigh>, <gasp>)",
         lines=5,
         placeholder=SAMPLES[0],
     ),
     outputs=gr.Audio(type="numpy", label="Generated Audio"),
     title="Orpheus‑3B Expressive TTS",
     description=(
-        "Use expressive tags **inside the text**.\n\n"
-        "Examples:\n"
-        "- `<laughs>`\n"
-        "- `<sighs>`\n"
-        "- `<whispers>`\n"
-        "- `<gasps>`\n"
-        "- `<nervous laughter>`\n\n"
-        "Tags can appear at the **start, middle, or end** of a sentence."
+        "You can add expressiveness to speech by inserting tags:\n"
+        "- <laugh>, <chuckle>: For laughter sounds\n"
+        "- <sigh>: For sighing sounds\n"
+        "- <cough>, <sniffle>: For subtle interruptions\n"
+        "- <groan>, <yawn>, <gasp>: For additional emotional expression\n\n"
+        "Tags can appear at the start, middle, or end of a sentence.\n"
+        "Example: `I can't believe it! <laugh>`"
     ),
     examples=[[s] for s in SAMPLES],
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(ssr_mode=False)