File size: 701 Bytes
06c46fb
 
 
 
 
 
 
 
 
 
 
 
 
1534a11
06c46fb
 
 
 
 
 
 
87e9667
 
e6e87d3
87e9667
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import os
import torch
from dotenv import load_dotenv

# Charger les variables d'environnement
load_dotenv()
HF_API_KEY = os.getenv("HF_API_KEY")

if not HF_API_KEY:
    raise ValueError("Le token Hugging Face n'a pas été trouvé dans .env")

# Labels d'émotions
LABELS = {"colere": 0, "neutre": 1, "joie": 2}
#LABELS = ["colere", "neutre", "joie"]
NUM_LABELS = len(LABELS)

# Choisir le device
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"

# Modèle Wav2Vec2
MODEL_NAME = "facebook/wav2vec2-large-xlsr-53-french"

BASE_DIR = os.path.dirname(os.path.abspath(__file__)) 
BEST_MODEL_NAME = os.path.join(BASE_DIR,"fr-speech-emotion-model.pth")  # Monte d'un niveau pour aller à la racine