fix app.py

app.py

@@ -5,22 +5,18 @@ from transformers import pipeline
 pipe_gtzan = pipeline(
     "audio-classification",
     model="sanchit-gandhi/distilhubert-finetuned-gtzan",
-    device="cpu",
 )
 
 pipe_esc = pipeline(
     "audio-classification",
     model="sonicase/distilhubert-finetuned-esc50",
-    device="cpu",
 )
 
 
 @spaces.GPU
 def classify(filepath):
-    pipe_gtzan.model.to("cuda")
-    pipe_esc.model.to("cuda")
-    genre = pipe_gtzan(filepath)
-    env = pipe_esc(filepath)
+    genre = pipe_gtzan(filepath, device="cuda")
+    env = pipe_esc(filepath, device="cuda")
     return (
         {p["label"]: p["score"] for p in genre},
         {p["label"]: p["score"] for p in env},