Spaces:
Runtime error
Runtime error
Vaibhav Srivastav
commited on
Commit
•
607a780
1
Parent(s):
5d0911d
yet another attempt at fixing the model_name
Browse files
app.py
CHANGED
@@ -59,7 +59,7 @@ def return_all_predictions(input_file, model_name):
|
|
59 |
|
60 |
|
61 |
gr.Interface(return_all_predictions,
|
62 |
-
inputs = [gr.inputs.Audio(source="microphone", type="filepath",
|
63 |
outputs = [gr.outputs.Textbox(label="Beam CTC Decoding"), gr.outputs.Textbox(label="Greedy Decoding")],
|
64 |
title="ASR using Wav2Vec 2.0 & pyctcdecode",
|
65 |
description = "Extending HF ASR models with pyctcdecode decoder",
|
|
|
59 |
|
60 |
|
61 |
gr.Interface(return_all_predictions,
|
62 |
+
inputs = [gr.inputs.Audio(source="microphone", type="filepath", label="Record/ Drop audio"), gr.inputs.Dropdown(["facebook/wav2vec2-base-960h", "facebook/hubert-large-ls960-ft"], label="Model Name")],
|
63 |
outputs = [gr.outputs.Textbox(label="Beam CTC Decoding"), gr.outputs.Textbox(label="Greedy Decoding")],
|
64 |
title="ASR using Wav2Vec 2.0 & pyctcdecode",
|
65 |
description = "Extending HF ASR models with pyctcdecode decoder",
|