Spaces:

badrex
/

arabic-dialect-identifier-demo

Running on Zero

badrex commited on Jun 23

Commit

7082254

verified ·

1 Parent(s): a6158c1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ print("Loading model...")
 model_id = "badrex/mms-300m-arabic-dialect-identifier"
 classifier = pipeline("audio-classification", model=model_id)
 print("Model loaded successfully")
-classifier.to('cuda')
 print("Model moved to GPU successfully")

 model_id = "badrex/mms-300m-arabic-dialect-identifier"
 classifier = pipeline("audio-classification", model=model_id)
 print("Model loaded successfully")
+classifier.model.to('cuda')
 print("Model moved to GPU successfully")