guangyil commited on
Commit
d66fc7f
·
verified ·
1 Parent(s): 2e6c9fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -27,6 +27,10 @@ from huggingface_hub import hf_hub_download
27
  from infer import load_model, eval_model
28
  from spkr import SpeakerEmbedding
29
 
 
 
 
 
30
 
31
  spkr_model = SpeakerEmbedding(device="cuda")
32
  model, tokenizer, tokenizer_voila, model_type = load_model("maitrix-org/Voila-chat", "maitrix-org/Voila-Tokenizer")
 
27
  from infer import load_model, eval_model
28
  from spkr import SpeakerEmbedding
29
 
30
+ @spaces.GPU
31
+ def dummy_gpu_check():
32
+ # Just to let Hugging Face Spaces know this app uses GPU
33
+ return torch.cuda.is_available()
34
 
35
  spkr_model = SpeakerEmbedding(device="cuda")
36
  model, tokenizer, tokenizer_voila, model_type = load_model("maitrix-org/Voila-chat", "maitrix-org/Voila-Tokenizer")