update_model_address

transcription/transcription.py

@@ -15,7 +15,7 @@ from .whisperx.alignment import load_align_model, align
 
 
 class MazeWhisperModel:
-    def __init__(self, model_name: str = "Sven33/maze-whisper-3000", device: str = "cuda"):
+    def __init__(self, model_name: str = "sven33/maze-whisper-3000", device: str = "cuda"):
         self.device = device
         self.model_name = model_name
         
@@ -50,7 +50,7 @@ class MazeWhisperModel:
 
 
 class WhisperXPipeline:
-    def __init__(self, model_name: str = "Sven33/maze-whisper-3000", device: str = "cuda", 
+    def __init__(self, model_name: str = "sven33/maze-whisper-3000", device: str = "cuda", 
                  vad_method: str = "pyannote", chunk_size: int = 30, 
                  enable_alignment: bool = True, align_language: str = "en"):
         self.device = device
@@ -288,7 +288,7 @@ def translate_audio_file(model: str = "mazeWhisper", audio_path: str = "", devic
     
     try:
         pipeline = WhisperXPipeline(
-            model_name="Sven33/maze-whisper-3000",
+            model_name="sven33/maze-whisper-3000",
             device=device,
             vad_method="pyannote",
             chunk_size=10,
@@ -356,7 +356,7 @@ def translate_audio_file(model: str = "mazeWhisper", audio_path: str = "", devic
         result_data = {
             "session_id": session_id,
             "audio_path": audio_path,
-            "model": "Sven33/maze-whisper-3000",
+            "model": "sven33/maze-whisper-3000",
             "device": device,
             "alignment_enabled": enable_alignment,
             "has_word_timestamps": has_word_timestamps,