EricChang
/

openai-whisper-large-v2-Lora

Automatic Speech Recognition

Model card Files Files and versions Community

cathyi commited on Jun 25, 2023

Commit

93e0850

·

1 Parent(s): c04524d

update model

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -12,7 +12,7 @@ from peft import LoraConfig, PeftModel, LoraModel, LoraConfig, get_peft_model, P
 class EndpointHandler():
     def __init__(self, path=""):
         # Preload all the elements you are going to need at inference.
-        peft_model_id = "cathyi/openai-whisper-large-v2-Lora"
         language = "Chinese"
         task = "transcribe"
         peft_config = PeftConfig.from_pretrained(peft_model_id)

 class EndpointHandler():
     def __init__(self, path=""):
         # Preload all the elements you are going to need at inference.
+        peft_model_id = "cathyi/tw-tw-openai-whisper-large-v2-Lora-epoch5-total5epoch"
         language = "Chinese"
         task = "transcribe"
         peft_config = PeftConfig.from_pretrained(peft_model_id)