Spaces:
Sleeping
Sleeping
Commit
·
02ad7fc
1
Parent(s):
04c3c08
Update app.py
Browse files
app.py
CHANGED
@@ -8,8 +8,13 @@ processor = WhisperProcessor.from_pretrained(model_id)
|
|
8 |
model = WhisperForConditionalGeneration.from_pretrained(model_id)
|
9 |
model.config.forced_decoder_ids = None
|
10 |
|
11 |
-
|
12 |
-
|
|
|
|
|
|
|
|
|
|
|
13 |
# Use the processor to transcribe the audio
|
14 |
transcription = processor.transcribe(audio, 48)
|
15 |
|
@@ -25,7 +30,7 @@ def transcribe(audio):
|
|
25 |
|
26 |
# Create a Gradio interface with two modes: realtime and file upload
|
27 |
iface = gr.Interface(
|
28 |
-
fn=
|
29 |
inputs=[
|
30 |
gr.inputs.Audio(source="microphone", type="numpy", label="Realtime Mode"),
|
31 |
gr.inputs.Audio(source="upload", type="numpy", label="File Upload Mode")
|
|
|
8 |
model = WhisperForConditionalGeneration.from_pretrained(model_id)
|
9 |
model.config.forced_decoder_ids = None
|
10 |
|
11 |
+
def transcribelocal(microphone, file_upload):
|
12 |
+
# Check which input is not None
|
13 |
+
if microphone is not None:
|
14 |
+
audio = microphone
|
15 |
+
else:
|
16 |
+
audio = file_upload
|
17 |
+
|
18 |
# Use the processor to transcribe the audio
|
19 |
transcription = processor.transcribe(audio, 48)
|
20 |
|
|
|
30 |
|
31 |
# Create a Gradio interface with two modes: realtime and file upload
|
32 |
iface = gr.Interface(
|
33 |
+
fn=transcribelocal,
|
34 |
inputs=[
|
35 |
gr.inputs.Audio(source="microphone", type="numpy", label="Realtime Mode"),
|
36 |
gr.inputs.Audio(source="upload", type="numpy", label="File Upload Mode")
|