Update app.py

app.py

@@ -1,9 +1,16 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, Speech2TextProcessor, Speech2TextForConditionalGeneration, VitsProcessor, VitsForConditionalGeneration
+from nemo.collections.asr.models import EncDecMultiTaskModel
 
-# Load the ASR model and processor
+# Load the ASR model and processor // fix processor stuff first
 asr_processor = Speech2TextProcessor.from_pretrained("/path/to/canary/processor")
-asr_model = Speech2TextForConditionalGeneration.from_pretrained("/path/to/canary/model")
+asr_model = EncDecMultiTaskModel.from_pretrained('nvidia/canary-1b')
+
+# update dcode params
+decode_cfg = canary_model.cfg.decoding
+decode_cfg.beam.beam_size = 1
+canary_model.change_decoding_strategy(decode_cfg)
+
 
 # Load the text processing model and tokenizer
 proc_tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-128k-instruct")