Spaces:

Marvis-AI
/

Marvis-TTS-250m

Running on Zero

mrfakename commited on 11 days ago

Commit

73b594a

verified ·

1 Parent(s): 784083a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 processor = AutoProcessor.from_pretrained(model_id)
 model = CsmForConditionalGeneration.from_pretrained(model_id, device_map=device)
-# TTS function
 def tts(text):
     inputs = processor(
         text,

 processor = AutoProcessor.from_pretrained(model_id)
 model = CsmForConditionalGeneration.from_pretrained(model_id, device_map=device)
+@spaces.GPU
 def tts(text):
     inputs = processor(
         text,