diff --git a/src/whisper_streaming/online_asr.py b/src/whisper_streaming/online_asr.py index dc34fd8..4d4b933 100644 --- a/src/whisper_streaming/online_asr.py +++ b/src/whisper_streaming/online_asr.py @@ -311,7 +311,7 @@ class VACOnlineASRProcessor(OnlineASRProcessor): import torch model, _ = torch.hub.load(repo_or_dir="snakers4/silero-vad", model="silero_vad") - from silero_vad_iterator import FixedVADIterator + from src.whisper_streaming.silero_vad_iterator import FixedVADIterator self.vac = FixedVADIterator( model diff --git a/silero_vad_iterator.py b/src/whisper_streaming/silero_vad_iterator.py similarity index 100% rename from silero_vad_iterator.py rename to src/whisper_streaming/silero_vad_iterator.py diff --git a/whisper_online.py b/src/whisper_streaming/whisper_online.py similarity index 100% rename from whisper_online.py rename to src/whisper_streaming/whisper_online.py diff --git a/whisper_fastapi_online_server.py b/whisper_fastapi_online_server.py index 9f84421..2cde185 100644 --- a/whisper_fastapi_online_server.py +++ b/whisper_fastapi_online_server.py @@ -9,7 +9,8 @@ from fastapi import FastAPI, WebSocket, WebSocketDisconnect from fastapi.responses import HTMLResponse from fastapi.middleware.cors import CORSMiddleware -from whisper_online import backend_factory, online_factory, add_shared_args +from src.whisper_streaming.whisper_online import backend_factory, online_factory, add_shared_args + app = FastAPI() app.add_middleware( CORSMiddleware,