pierreguillou commited on
Commit
bb459b1
·
verified ·
1 Parent(s): 9805c24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,7 +7,8 @@ from langdetect import detect
7
  from langdetect.lang_detect_exception import LangDetectException
8
 
9
  # --- Configuration ---
10
- LANG_MODEL_NAME = "openai/whisper-tiny" # modèle léger pour la détection de langue
 
11
 
12
  device = 0 if torch.cuda.is_available() else "cpu"
13
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
7
  from langdetect.lang_detect_exception import LangDetectException
8
 
9
  # --- Configuration ---
10
+ #LANG_MODEL_NAME = "openai/whisper-tiny" # modèle léger pour la détection de langue
11
+ LANG_MODEL_NAME = "openai/whisper-base" # modèle medium pour une meilleure détection de langue
12
 
13
  device = 0 if torch.cuda.is_available() else "cpu"
14
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32