Update app.py
Browse files
app.py
CHANGED
|
@@ -134,14 +134,14 @@ def yt_transcribe(yt_url, task, use_demucs, dataset_name, oauth_token: gr.OAuthT
|
|
| 134 |
with open(filepath, "rb") as f:
|
| 135 |
inputs_path = f.read()
|
| 136 |
|
| 137 |
-
|
| 138 |
-
|
| 139 |
-
|
| 140 |
-
out = pipe(inputs, batch_size=BATCH_SIZE, generate_kwargs={"task": task}, return_timestamps=True)
|
| 141 |
|
| 142 |
-
|
| 143 |
|
| 144 |
-
|
|
|
|
|
|
|
| 145 |
|
| 146 |
chunks = naive_postprocess_whisper_chunks(out["chunks"], inputs, dataset_sampling_rate)
|
| 147 |
|
|
|
|
| 134 |
with open(filepath, "rb") as f:
|
| 135 |
inputs_path = f.read()
|
| 136 |
|
| 137 |
+
inputs = ffmpeg_read(inputs_path, pipe.feature_extractor.sampling_rate)
|
| 138 |
+
inputs = {"array": inputs, "sampling_rate": pipe.feature_extractor.sampling_rate}
|
|
|
|
|
|
|
| 139 |
|
| 140 |
+
out = pipe(inputs, batch_size=BATCH_SIZE, generate_kwargs={"task": task}, return_timestamps=True)
|
| 141 |
|
| 142 |
+
text = out["text"]
|
| 143 |
+
|
| 144 |
+
inputs = ffmpeg_read(inputs_path, dataset_sampling_rate)
|
| 145 |
|
| 146 |
chunks = naive_postprocess_whisper_chunks(out["chunks"], inputs, dataset_sampling_rate)
|
| 147 |
|