Update app.py

app.py

@@ -10,14 +10,17 @@ model_id = os.getenv("MODEL_ID")
 
 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
-model = Wav2Vec2ForCTC.from_pretrained(model_id, target_lang="khm", token=hf_token, low_cpu_mem_usage=True, dtype=torch_dtype).to("cpu")
-processor = Wav2Vec2Processor.from_pretrained(model_id, target_lang="khm", token=hf_token)
+#model = Wav2Vec2ForCTC.from_pretrained(model_id, target_lang="khm", token=hf_token, low_cpu_mem_usage=True, dtype=torch_dtype).to("cpu")
+#processor = Wav2Vec2Processor.from_pretrained(model_id, target_lang="khm", token=hf_token)
 
 pipe = pipeline(
-    "automatic-speech-recognition",
-    model=model.eval(),
-    tokenizer=processor.tokenizer,
-    feature_extractor=processor.feature_extractor,
+    #"automatic-speech-recognition",
+    #model=model.eval(),
+    #tokenizer=processor.tokenizer,
+    #feature_extractor=processor.feature_extractor,
+    model=model_id,
+    model_kwargs={"target_lang": "khm", "ignore_mismatched_sizes": True},
+    token=hf_token,
     torch_dtype=torch_dtype,
     device_map="cpu"