edm_genre_classificator

Sleeping

jvalero commited on Jul 15

Commit

158b7b0

•

1 Parent(s): 804f975

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import gradio as gr
 device = 0 if torch.cuda.is_available() else "cpu"
-MODEL_ID = "wav2vec2-base-vinyl_condition"
 pipe = pipeline(
     task="audio-classification",

 device = 0 if torch.cuda.is_available() else "cpu"
+MODEL_ID = "jvalero/wav2vec2-base-vinyl_condition"
 pipe = pipeline(
     task="audio-classification",