rahul7star commited on
Commit
fc72e9f
·
verified ·
1 Parent(s): f036d34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -3
app.py CHANGED
@@ -31,15 +31,26 @@ class TTSRequest(BaseModel):
31
  # -------------------------------------------------
32
  # Model loader (NO .eval())
33
  # -------------------------------------------------
 
 
34
  def get_or_load_model():
35
  global MODEL
 
36
  if MODEL is None:
37
- MODEL = ChatterboxMultilingualTTS(
38
- device="cuda" if torch.cuda.is_available() else "cpu"
39
- )
 
 
 
 
 
 
 
40
  return MODEL
41
 
42
 
 
43
  # -------------------------------------------------
44
  # API: TTS
45
  # -------------------------------------------------
@@ -97,3 +108,4 @@ def ui():
97
  @app.on_event("startup")
98
  def warmup():
99
  get_or_load_model()
 
 
31
  # -------------------------------------------------
32
  # Model loader (NO .eval())
33
  # -------------------------------------------------
34
+ MODEL = None
35
+
36
  def get_or_load_model():
37
  global MODEL
38
+
39
  if MODEL is None:
40
+ print("🔄 Loading ChatterboxMultilingualTTS (CPU-only)")
41
+
42
+ # ✅ THIS is the ONLY valid loader
43
+ MODEL = ChatterboxMultilingualTTS.from_pretrained("cpu")
44
+
45
+ # Chatterbox is NOT torch.nn.Module → no .to()
46
+ MODEL.eval()
47
+
48
+ print("✅ Chatterbox model loaded successfully")
49
+
50
  return MODEL
51
 
52
 
53
+
54
  # -------------------------------------------------
55
  # API: TTS
56
  # -------------------------------------------------
 
108
  @app.on_event("startup")
109
  def warmup():
110
  get_or_load_model()
111
+