Update apis/chat_api.py

apis/chat_api.py

@@ -155,15 +155,15 @@ class ChatAPIApp:
         audio_file: UploadFile = File(
             description="Source Audio File",
         )
-    def whisper_transcribe(self, item:WhisperPostItem):
+    def whisper_transcribe(self, model:str = Field(default="whisper-small",description="(str) `Whisper model`"), lang:str = Field(default="en",description="(str) `transcribe to`"), audio_file: UploadFile = File(description="Source Audio File")):
         MODEL_MAP = {
         "whisper-small": "openai/whisper-small",
         "whisper-medium": "openai/whisper-medium",
         "whisper-large": "openai/whisper-large",   
         "default": "openai/whisper-small",
         }
-        if item.model in MODEL_MAP.keys():
-            target_model = item.model
+        if model in MODEL_MAP.keys():
+            target_model = model
         else:
             target_model = "default"
 
@@ -176,8 +176,8 @@ class ChatAPIApp:
            device=device,
         )
         time_start = time.time()
-        pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language=item.lang, task="transcribe")
-        text = pipe(item.audio_file)["text"]
+        pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language=lang, task="transcribe")
+        text = pipe(audio_file)["text"]
         time_end = time.time()
         item_response = {
             "statue": 200,