DarwinAnim8or commited on
Commit
02ad7fc
·
1 Parent(s): 04c3c08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
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
- # Define a function that takes an audio input and returns a transcription
12
- def transcribe(audio):
 
 
 
 
 
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=transcribe,
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")