Spaces:
Runtime error
Runtime error
Fix: Configuration HF Spaces dans README.md
Browse files- app.py +3 -2
- app_with_api.py +3 -2
app.py
CHANGED
@@ -17,12 +17,13 @@ HF_SPACE = os.getenv("HF_SPACE", "false").lower() == "true"
|
|
17 |
|
18 |
# Préchargement des modèles
|
19 |
print("Chargement des modèles...")
|
|
|
20 |
processor_ctc = Wav2Vec2Processor.from_pretrained(
|
21 |
-
"
|
22 |
cache_dir="./models" if not HF_SPACE else None
|
23 |
)
|
24 |
model_ctc = Wav2Vec2ForCTC.from_pretrained(
|
25 |
-
"
|
26 |
cache_dir="./models" if not HF_SPACE else None
|
27 |
)
|
28 |
|
|
|
17 |
|
18 |
# Préchargement des modèles
|
19 |
print("Chargement des modèles...")
|
20 |
+
# Modèle français plus léger
|
21 |
processor_ctc = Wav2Vec2Processor.from_pretrained(
|
22 |
+
"LeBenchmark/wav2vec2-FR-2K-small",
|
23 |
cache_dir="./models" if not HF_SPACE else None
|
24 |
)
|
25 |
model_ctc = Wav2Vec2ForCTC.from_pretrained(
|
26 |
+
"LeBenchmark/wav2vec2-FR-2K-small",
|
27 |
cache_dir="./models" if not HF_SPACE else None
|
28 |
)
|
29 |
|
app_with_api.py
CHANGED
@@ -25,12 +25,13 @@ HF_SPACE = os.getenv("HF_SPACE", "false").lower() == "true"
|
|
25 |
|
26 |
# Préchargement des modèles (partagés entre Gradio et API)
|
27 |
print("Chargement des modèles...")
|
|
|
28 |
processor_ctc = Wav2Vec2Processor.from_pretrained(
|
29 |
-
"
|
30 |
cache_dir="./models" if not HF_SPACE else None
|
31 |
)
|
32 |
model_ctc = Wav2Vec2ForCTC.from_pretrained(
|
33 |
-
"
|
34 |
cache_dir="./models" if not HF_SPACE else None
|
35 |
)
|
36 |
|
|
|
25 |
|
26 |
# Préchargement des modèles (partagés entre Gradio et API)
|
27 |
print("Chargement des modèles...")
|
28 |
+
# Modèle français plus léger
|
29 |
processor_ctc = Wav2Vec2Processor.from_pretrained(
|
30 |
+
"LeBenchmark/wav2vec2-FR-2K-small",
|
31 |
cache_dir="./models" if not HF_SPACE else None
|
32 |
)
|
33 |
model_ctc = Wav2Vec2ForCTC.from_pretrained(
|
34 |
+
"LeBenchmark/wav2vec2-FR-2K-small",
|
35 |
cache_dir="./models" if not HF_SPACE else None
|
36 |
)
|
37 |
|