Update app.py
Browse files
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
|