Spaces:

abidlabs
/

speech-translation

Build error

patrickvonplaten commited on Dec 16, 2021

Commit

ad145b5

•

1 Parent(s): d8a335e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ def process_audio_file(file):
     input_values = feature_extractor(data, return_tensors="pt").input_values.to(device)
     if torch.cuda.is_available():
-        input_values.to(torch.float16)
     return input_values
 def transcribe(file_mic, file_upload, target_language):

     input_values = feature_extractor(data, return_tensors="pt").input_values.to(device)
     if torch.cuda.is_available():
+        input_values = input_values.to(torch.float16)
     return input_values
 def transcribe(file_mic, file_upload, target_language):