Added spaces zerogpu

xtts.py

@@ -7,6 +7,7 @@ import torch
 import numpy as np
 from typing import List
 from pydantic import BaseModel
+import spaces
 
 from TTS.tts.configs.xtts_config import XttsConfig
 from TTS.tts.models.xtts import Xtts
@@ -145,6 +146,7 @@ class TTSInputs(BaseModel):
     top_p: float
 
 # @app.post("/tts")
++@spaces.GPU
 def predict_speech(parsed_input: TTSInputs):
     speaker_embedding = torch.tensor(parsed_input.speaker_embedding).unsqueeze(0).unsqueeze(-1)
     gpt_cond_latent = torch.tensor(parsed_input.gpt_cond_latent).reshape((-1, 1024)).unsqueeze(0)