Spaces:

badrex
/

JASRv1.1

Running on Zero

badrex commited on Oct 20

Commit

4e5901e

verified ·

1 Parent(s): b62fc3c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ def process_audio(audio_path):
     #audio_array = {k: v.to(device) for k, v in inputs.items()}
     #audio_array = audio_array.to(device)
-    inputs = inputs = processor(audio_array.cpu().numpy(), sampling_rate=16000, return_tensors="pt")
     inputs = inputs.to(device, dtype=torch.bfloat16)
     with torch.no_grad():

     #audio_array = {k: v.to(device) for k, v in inputs.items()}
     #audio_array = audio_array.to(device)
+    inputs = processor(audio_array, sampling_rate=16000, return_tensors="pt")
     inputs = inputs.to(device, dtype=torch.bfloat16)
     with torch.no_grad():