Karayakar commited on
Commit
c0ae2ba
·
verified ·
1 Parent(s): 4897092

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ load_dotenv()
9
 
10
  # Check if CUDA is available
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
-
13
  print("Loading SNAC model...")
14
  snac_model = SNAC.from_pretrained("hubertsiuzdak/snac_24khz")
15
  snac_model = snac_model.to(device)
 
9
 
10
  # Check if CUDA is available
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
+ hf_token = os.environ["HF_TOKEN"]
13
  print("Loading SNAC model...")
14
  snac_model = SNAC.from_pretrained("hubertsiuzdak/snac_24khz")
15
  snac_model = snac_model.to(device)