Update app.py

app.py

@@ -5,9 +5,16 @@ import boto3
 import json
 import soundfile as sf
 import tempfile
-from generator import load_csm_1b, Segment
+from generator import load_csm_1b  # ✅ Only import the loader
 import whisper  # or use faster-whisper if preferred
 
+# ✅ Embedded Segment class here to fix import error
+class Segment:
+    def __init__(self, text, speaker=0, audio=None):
+        self.text = text
+        self.speaker = speaker
+        self.audio = audio
+
 # Load models once
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"Loading Sesame CSM model on {device}...")