Update app.py
Browse files
app.py
CHANGED
@@ -26,8 +26,9 @@ def transcribe(audio):
|
|
26 |
return "Espera a que la grabaci贸n termine de subirse al servidor !! Intentelo de nuevo en unos segundos"
|
27 |
|
28 |
sr, y = audio
|
29 |
-
y = y.astype(np.
|
30 |
y /= np.max(np.abs(y))
|
|
|
31 |
return pipeline({"sampling_rate": sr, "raw": y}, generate_kwargs={"forced_decoder_ids": forced_decoder_ids}, max_new_tokens=255)["text"]
|
32 |
|
33 |
examples = [
|
|
|
26 |
return "Espera a que la grabaci贸n termine de subirse al servidor !! Intentelo de nuevo en unos segundos"
|
27 |
|
28 |
sr, y = audio
|
29 |
+
y = y.astype(np.float32)
|
30 |
y /= np.max(np.abs(y))
|
31 |
+
y = torch.from_numpy(y).to(torch.float16).to('cuda')
|
32 |
return pipeline({"sampling_rate": sr, "raw": y}, generate_kwargs={"forced_decoder_ids": forced_decoder_ids}, max_new_tokens=255)["text"]
|
33 |
|
34 |
examples = [
|