Update apis/chat_api.py

apis/chat_api.py

@@ -153,9 +153,9 @@ class ChatAPIApp:
             description="(str) `transcribe to`",
         )
         audio_file: UploadFile = File(
-            description="Source Audio File",
+            description="(File) `Source Audio File`",
         )
-    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")):
+    def whisper_transcribe(self, item:WhisperPostItem):
         MODEL_MAP = {
         "whisper-small": "openai/whisper-small",
         "whisper-medium": "openai/whisper-medium",
@@ -163,7 +163,7 @@ class ChatAPIApp:
         "default": "openai/whisper-small",
         }
         if model in MODEL_MAP.keys():
-            target_model = model
+            target_model = item.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=lang, task="transcribe")
-        text = pipe(audio_file)["text"]
+        pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language=item.lang, task="transcribe")
+        text = pipe(item.audio_file)["text"]
         time_end = time.time()
         item_response = {
             "statue": 200,