Spaces:

Elormiden
/

Simona-AI

Running

Elormiden commited on May 28

Commit

fb02cb1

verified ·

1 Parent(s): 02f9259

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ ASR = pipeline(
     alpha=0.4,
     beta=0.9,
     token=HF_TOKEN,
-    device=0,
     model_id_or_path=MODEL_ID,
 )

     alpha=0.4,
     beta=0.9,
     token=HF_TOKEN,
+    device = 0 if torch.cuda.is_available() else -1,
     model_id_or_path=MODEL_ID,
 )