maxmustermann/NeMo
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Update the webapp for ASR (#3032)

Browse source 
Signed-off-by: smajumdar <titu1994@gmail.com>
This commit is contained in:
Somshubra Majumdar 2021-10-21 10:30:45 -07:00 committed by GitHub
parent 32fa5cfaf3
commit 4e544676f2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
nemo/collections/asr/data/audio_to_text_dali.py
View file

@ -110,7 +110,6 @@ class DALIOutputs(object):
return len(self._outs)
@experimental
class _AudioTextDALIDataset(Iterator):
"""
NVIDIA DALI pipeline that loads tensors via one or more manifest files where each line containing a sample descriptor in JSON,

2
tools/asr_webapp/Dockerfile
View file

@ -14,7 +14,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
ARG BASE_IMAGE=nvcr.io/nvidia/nemo:1.0.1
ARG BASE_IMAGE=nvcr.io/nvidia/nemo:1.4.0
# build an image that includes only the nemo dependencies, ensures that dependencies
# are included first for optimal caching, and useful for building a development

2
tools/asr_webapp/asr_service.py
View file

@ -371,4 +371,4 @@ atexit.register(remove_tmp_dir_at_exit)
if __name__ == '__main__':
app.run(debug=True)
app.run(False)

4
tools/asr_webapp/model_api.py
View file

@ -48,8 +48,10 @@ def get_model_names():
for model_info in nemo_asr.models.ASRModel.list_available_models():
for superclass in model_info.class_.mro():
if 'CTC' in superclass.__name__ or 'RNNT' in superclass.__name__:
if 'align' in model_info.pretrained_model_name:
continue
nemo_model_names.add(model_info.pretrained_model_name)
break
nemo_model_names = list(sorted(nemo_model_names))
return nemo_model_names, local_model_names