Update app.py

app.py

@@ -7,6 +7,7 @@ from huggingface_hub import model_info
 import openai
 
 
+
 def transcribe(microphone, file_upload):
     warn_output = ""
     if (microphone is not None) and (file_upload is not None):
@@ -20,6 +21,13 @@ def transcribe(microphone, file_upload):
 
     file = microphone if microphone is not None else file_upload
 
+    openai.api_key = "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"
+    device = 0 if torch.cuda.is_available() else "cpu"
+    res_format = 'srt'
+    pipe = openai.Audio.transcribe(model="whisper-1", file=open('file', 'rb'), response_format=res_format, prompt='请使用书面语')
+    pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language=lang, task="transcribe")
+
+
     text = pipe(file)["text"]
 
     return warn_output + text
@@ -45,7 +53,7 @@ def yt_transcribe(yt_url):
     res_format = 'srt'
     pipe = openai.Audio.transcribe(model="whisper-1", file=open('audio.mp3', 'rb'), response_format=res_format, prompt='请使用书面语')
     pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language=lang, task="transcribe")
-
+    
     text = pipe("audio.mp3")["text"]
 
     return html_embed_str, text