Update asr.py

doppeltilde · Jun 13, 2024 · 085c661 · 085c661
1 parent b8063f0
commit 085c661
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/routes/api/asr.py b/src/routes/api/asr.py
@@ -5,6 +5,8 @@
 import traceback
 import time
 from src.shared.shared import LANGUAGE_CODES
+import gc
+import torch
 
 router = APIRouter()
 
@@ -67,3 +69,6 @@ async def asr(
 
     finally:
         audio_file.close()
+        del model
+        torch.cuda.empty_cache()
+        gc.collect()