Merge branch 'feature/local-asr-demo'

pyproject.toml

@@ -4,4 +4,7 @@ version = "0.1.0"
 description = "Huggingface Space to test NVIDIA Nemo's Parakeet v2 speech recognition model."
 readme = "README.md"
 requires-python = ">=3.10"
-dependencies = []
+dependencies = [
+    "nemo-toolkit[asr]>=2.2.1",
+    "numpy<2.0",
+]

run.py

@@ -1,6 +1,16 @@
-def main():
-    print("Hello from parakeet-tdt-0-6b-v2!")
+import sys
+
+import nemo.collections.asr as nemo_asr
+
+
+def main(argv):
+    model = nemo_asr.models.ASRModel.from_pretrained(
+        model_name="nvidia/parakeet-tdt-0.6b-v2"
+    )
+
+    output = model.transcribe([argv[1]])
+    print(output[0].text)
 
 
 if __name__ == "__main__":
-    main()
+    main(sys.argv)