Spaces:
Sleeping
Sleeping
Sofia Casadei
commited on
Commit
Β·
edfee48
1
Parent(s):
0fa1945
fix use flash attn
Browse files
main.py
CHANGED
@@ -91,8 +91,8 @@ async def transcribe(audio: tuple[int, np.ndarray]):
|
|
91 |
logger.info(f"Sample rate: {sample_rate}Hz, Shape: {audio_array.shape}")
|
92 |
|
93 |
outputs = transcribe_pipeline(
|
94 |
-
|
95 |
-
audio_array, # pass numpy array
|
96 |
chunk_length_s=3,
|
97 |
batch_size=1,
|
98 |
generate_kwargs={
|
|
|
91 |
logger.info(f"Sample rate: {sample_rate}Hz, Shape: {audio_array.shape}")
|
92 |
|
93 |
outputs = transcribe_pipeline(
|
94 |
+
audio_to_bytes(audio), # pass bytes
|
95 |
+
#audio_array, # pass numpy array
|
96 |
chunk_length_s=3,
|
97 |
batch_size=1,
|
98 |
generate_kwargs={
|