Update transcribe.py

transcribe.py

@@ -4,7 +4,10 @@ from typing import Dict
 import nemo.collections.asr as nemo_asr
 
 
-def predict_model(model_path: str, audio_file_path: str) -> Dict:
+def predict_model(
+        model_path: str = None,
+        audio_file_path: str = None,
+) -> Dict:
     # Restore the ASR model from the provided path
     model = nemo_asr.models.ASRModel.restore_from(restore_path=model_path)
     # Transcribe the given audio file
@@ -16,7 +19,10 @@ if __name__ == "__main__":
     # Parse command line arguments
     parser = argparse.ArgumentParser()
     parser.add_argument("--model_path", default=None, help="Path to a model to evaluate.")
-    parser.add_argument("--audio_file_path", help="Path for train manifest JSON file.")
+    parser.add_argument("--audio_file_path", default=None, help="Path for train manifest JSON file.")
     args = parser.parse_args()
 
-    predict_model(model_path=args.model_path, audio_file_path=args.audio_file_path)
+    predict_model(
+        model_path=args.model_path, 
+        audio_file_path=args.audio_file_path,
+    )