Spaces:

Pattr
/

DrumClassification

Sleeping

Pattr commited on May 25, 2023

Commit

5b50e3c

1 Parent(s): 8dee12a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,8 +5,8 @@ from pydub import *
 import fastbook
 from fastbook import *
 import pathlib
-model = load_learner("thedrum.pkl")
 def play(x,y,ti,beat,audio):

 import fastbook
 from fastbook import *
 import pathlib
+path = Path()
+model = load_learner(path/"thedrum.pkl", cpu=True)
 def play(x,y,ti,beat,audio):