Update app.py

app.py

@@ -6,7 +6,7 @@ from huggingface_hub import snapshot_download
 from omegaconf import OmegaConf
 from nemo.collections.asr.models import ASRModel
 
-# Use the correct repository name
+# Pointing to your cloned model repository
 MODEL_NAME = "trysem/stt_ml_fastconformer_ctc_med_punct"
 
 print("1. Downloading unzipped model files from Hugging Face...")
@@ -24,11 +24,18 @@ if 'encoder' in config:
     config.encoder.pop('use_pytorch_sdpa', None)
     config.encoder.pop('use_pytorch_sdpa_backends', None)
 
-# Patch 2: Downgrade 'greedy_batch' strategy to 'greedy'
+# Patch 2 & 3: Fix decoding strategy and confidence config
 if 'decoding' in config:
+    # Patch 2: Downgrade 'greedy_batch' strategy to 'greedy'
     if config.decoding.get('strategy') == 'greedy_batch':
         print("   -> Downgrading decoding strategy from 'greedy_batch' to 'greedy'")
         config.decoding.strategy = 'greedy'
+        
+    # Patch 3: Remove 'tdt_include_duration' which NeMo 1.23 doesn't recognize
+    if 'confidence_cfg' in config.decoding and config.decoding.confidence_cfg is not None:
+        if 'tdt_include_duration' in config.decoding.confidence_cfg:
+            print("   -> Removing 'tdt_include_duration' from confidence_cfg")
+            config.decoding.confidence_cfg.pop('tdt_include_duration', None)
 
 print("3. Packaging files into a standard .nemo archive...")
 patched_dir = "patched_nemo_env"