Upload 3 files

Dockerfile

@@ -12,4 +12,4 @@ RUN pip install --no-cache-dir -r requirements.txt
 COPY . .
 
 # サーバーを起動するコマンドを指定
-CMD ["python", "./whisper_fastapi_online_server.py", "--host", "localhost", "--port", "8000", "--lan", "en", "--model", "tiny"]
+CMD ["python", "./whisper_fastapi_online_server.py", "--host", "0.0.0.0", "--port", "8000", "--lan", "en", "--model", "tiny"]

requirements.txt

@@ -2,4 +2,6 @@ librosa
 soundfile
 fastapi
 ffmpeg-python
-faster-whisper
+faster-whisper
+transformers
+numpy

whisper_fastapi_online_server.py

@@ -1,3 +1,4 @@
+import os
 import io
 import requests
 import argparse
@@ -12,7 +13,6 @@ from fastapi.middleware.cors import CORSMiddleware
 
 from src.whisper_streaming.whisper_online import backend_factory, online_factory, add_shared_args
 
-
 import logging
 import logging.config
 from transformers import pipeline
@@ -21,7 +21,7 @@ MODEL_NAME = 'Helsinki-NLP/opus-tatoeba-en-ja'
 TRANSLATOR = pipeline('translation', model=MODEL_NAME, device='cuda')
 TRANSLATOR('Warming up!')
 
-API_KEY = '3c2b8b0f-4fa9-4eb7-b67d-7cae25546051:fx' # 自身の API キーを指定
+DEEPL_API_KEY = os.environ['DEEPL_API_KEY']
 
 SOURCE_LANG = 'EN'
 TARGET_LANG = 'JA'
@@ -29,7 +29,7 @@ TARGET_LANG = 'JA'
 def translator_wrapper(source_text, mode='deepl'):
     if mode == 'deepl':
         params = {
-                    'auth_key' : API_KEY,
+                    'auth_key' : DEEPL_API_KEY,
                     'text' : source_text,
                     'source_lang' : SOURCE_LANG, # 翻訳対象の言語
                     "target_lang": TARGET_LANG  # 翻訳後の言語