Spaces:

badrex
/

arabic-dialect-identifier-demo

Running on Zero

badrex commited on Jun 23

Commit

75f1597

verified ·

1 Parent(s): 7082254

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,9 +7,8 @@ import torch
 # Load the model
 print("Loading model...")
 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")

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