Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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")
|