Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -2,8 +2,7 @@ import gradio as gr
|
|
2 |
import torch
|
3 |
from transformers import pipeline
|
4 |
|
5 |
-
|
6 |
-
model_id = f"{username}/distilhubert-finetuned-gtzan"
|
7 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
8 |
pipe = pipeline("audio-classification", model=model_id, device=device)
|
9 |
|
|
|
2 |
import torch
|
3 |
from transformers import pipeline
|
4 |
|
5 |
+
model_id = "ntu-spml/distilhubert"
|
|
|
6 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
7 |
pipe = pipeline("audio-classification", model=model_id, device=device)
|
8 |
|