Update stitched_model.py

stitched_model.py

@@ -1,12 +1,12 @@
 import torch
 from torch import nn
-from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC, AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import AutoProcessor, Wav2Vec2ForCTC, AutoTokenizer, AutoModelForSeq2SeqLM
 
 class CombinedModel(nn.Module):
     def __init__(self, stt_model_name, nmt_model_name,device = "cuda"):
         super(CombinedModel, self).__init__()
 
-        self.stt_processor = Wav2Vec2Processor.from_pretrained(stt_model_name)
+        self.stt_processor = AutoProcessor.from_pretrained(stt_model_name)
         self.stt_model = Wav2Vec2ForCTC.from_pretrained(stt_model_name)
         self.nmt_tokenizer = AutoTokenizer.from_pretrained(nmt_model_name)
         self.nmt_model = AutoModelForSeq2SeqLM.from_pretrained(nmt_model_name)