Spaces:
Sleeping
Sleeping
tqdm?
Browse files
app.py
CHANGED
@@ -15,11 +15,11 @@ urllib.request.urlretrieve(
|
|
15 |
writer = whisper.utils.get_writer("srt", "/dev/null")
|
16 |
|
17 |
@spaces.GPU
|
18 |
-
def generate(file):
|
19 |
# get file to type bytes somehow
|
20 |
model = whisper.load_model("large", device="cuda")
|
21 |
audio = whisper.load_audio(file)
|
22 |
-
result = model.transcribe(audio)
|
23 |
out = io.StringIO()
|
24 |
writer.write_result(result, out)
|
25 |
return out.getvalue()
|
|
|
15 |
writer = whisper.utils.get_writer("srt", "/dev/null")
|
16 |
|
17 |
@spaces.GPU
|
18 |
+
def generate(file, progress=gr.Progress(track_tqdm=True)):
|
19 |
# get file to type bytes somehow
|
20 |
model = whisper.load_model("large", device="cuda")
|
21 |
audio = whisper.load_audio(file)
|
22 |
+
result = model.transcribe(audio, verbose=False)
|
23 |
out = io.StringIO()
|
24 |
writer.write_result(result, out)
|
25 |
return out.getvalue()
|