music-genre-classifier

Runtime error

smpol commited on Jan 9

Commit

94a728f

•

1 Parent(s): 0b7d49b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 from transformers import pipeline
 username = "ardneebwar"  ## Complete your username
-model_id = f"{username}/distilhubert-finetuned-gtzan"
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 pipe = pipeline("audio-classification", model=model_id, device=device)

 from transformers import pipeline
 username = "ardneebwar"  ## Complete your username
+model_id = "sanchit-gandhi/distilhubert-finetuned-gtzan"
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 pipe = pipeline("audio-classification", model=model_id, device=device)