Update app.py

app.py

@@ -25,7 +25,7 @@ from os import getenv
 from zonos.model import Zonos
 from zonos.conditioning import make_cond_dict, supported_language_codes
 
-device = "cuda"
+device = 0 if torch.cuda.is_available() else "cpu"
 MODEL_NAMES = ["Zyphra/Zonos-v0.1-transformer", "Zyphra/Zonos-v0.1-hybrid"]
 MODELS = {name: Zonos.from_pretrained(name, device=device) for name in MODEL_NAMES}
 for model in MODELS.values():