Spaces:

sparshmehta
/

main_app

Sleeping

sparshmehta commited on Feb 10, 2025

Commit

3289a6d

verified ·

1 Parent(s): 1c68fc6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -664,7 +664,7 @@ class MentorEvaluator:
                 logger.info("Attempting to initialize Whisper model...")
                 # First try to initialize model with downloading allowed
                 self._whisper_model = WhisperModel(
-                    "small",
                     device="cpu",
                     compute_type="int8",
                     download_root=self.model_cache_dir,
@@ -677,7 +677,7 @@ class MentorEvaluator:
                 try:
                     logger.info("Attempting to load model from local cache...")
                     self._whisper_model = WhisperModel(
-                        "small",
                         device="cpu",
                         compute_type="int8",
                         download_root=self.model_cache_dir,

                 logger.info("Attempting to initialize Whisper model...")
                 # First try to initialize model with downloading allowed
                 self._whisper_model = WhisperModel(
+                    "medium",
                     device="cpu",
                     compute_type="int8",
                     download_root=self.model_cache_dir,
                 try:
                     logger.info("Attempting to load model from local cache...")
                     self._whisper_model = WhisperModel(
+                        "medium",
                         device="cpu",
                         compute_type="int8",
                         download_root=self.model_cache_dir,