alec228 commited on
Commit
0986800
·
1 Parent(s): 96f87e4

Fix: Configuration HF Spaces dans README.md

Browse files
Files changed (2) hide show
  1. app.py +3 -2
  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
- "jonatasgrosman/wav2vec2-large-xlsr-53-french",
22
  cache_dir="./models" if not HF_SPACE else None
23
  )
24
  model_ctc = Wav2Vec2ForCTC.from_pretrained(
25
- "jonatasgrosman/wav2vec2-large-xlsr-53-french",
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
- "jonatasgrosman/wav2vec2-large-xlsr-53-french",
30
  cache_dir="./models" if not HF_SPACE else None
31
  )
32
  model_ctc = Wav2Vec2ForCTC.from_pretrained(
33
- "jonatasgrosman/wav2vec2-large-xlsr-53-french",
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