Refactor: Move device definitions to config file

lib/config.py

@@ -2,6 +2,7 @@
 # -*- coding: utf-8 -*-
 
 from pathlib import Path
+import torch
 
 # -->> Tunables <<---------------------
 
@@ -23,6 +24,16 @@ TRANSCRIPTS_DIR = MEDIA_DIR / "Transcriptions"
 AUDIO_EXT = ".mp3"
 TEXT_EXT = ".txt"
 
+# Model parameters
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+BATCH_SIZE = 16
+
+if DEVICE == "cuda":
+    COMPUTE_TYPE = "float16"
+else:
+    COMPUTE_TYPE = "int8"
+
+print("Using device:", DEVICE)
 # -->> API <<--------------------------
 
 

lib/model.py

@@ -6,17 +6,14 @@ from time import time
 import torch
 import whisperx as wx
 
+from .config import (
+    DEVICE,
+    COMPUTE_TYPE,
+    BATCH_SIZE,
+)
 # -->> Tunables <<---------------------
 
-DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
-BATCH_SIZE = 16
 
-if DEVICE == "cuda":
-    COMPUTE_TYPE = "float16"
-else:
-    COMPUTE_TYPE = "int8"
-
-print("Using device:", DEVICE)
 # -->> Definitions <<------------------