victormattosli commited on
Commit
154c369
·
verified ·
1 Parent(s): e78e314

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -1
handler.py CHANGED
@@ -6,6 +6,7 @@ import io
6
  import os
7
  import base64 # <-- Nova importação
8
  import matplotlib.pyplot as plt # <-- Nova importação
 
9
  from scipy.signal import butter, lfilter
10
  from transformers import pipeline, AutoImageProcessor, AutoModelForImageClassification
11
  from PIL import Image
@@ -82,7 +83,7 @@ class EndpointHandler:
82
 
83
  # Gera um buffer em memória com o áudio filtrado
84
  buffer = io.BytesIO()
85
- sf.write(buffer, y_cleaned, TARGET_SR, format='MP3')
86
  buffer.seek(0)
87
 
88
  # Codifica em Base64
 
6
  import os
7
  import base64 # <-- Nova importação
8
  import matplotlib.pyplot as plt # <-- Nova importação
9
+ import soundfile as sf
10
  from scipy.signal import butter, lfilter
11
  from transformers import pipeline, AutoImageProcessor, AutoModelForImageClassification
12
  from PIL import Image
 
83
 
84
  # Gera um buffer em memória com o áudio filtrado
85
  buffer = io.BytesIO()
86
+ sf.write(buffer, y_cleaned, TARGET_SR, format='WAV')
87
  buffer.seek(0)
88
 
89
  # Codifica em Base64