Hugging Face's logo

Spaces:
STT-Darija-ORG
/
hadra-api
Sleeping

App Files Community
Ala404 commited on
Commit
9ff04de
·
verified ·
1 Parent(s): 8c12a1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ app.add_middleware(
28
  )
29
 
30
  # Load the pipeline with device set to GPU if available
31
- MODEL_NAME = "STT-Darija-ORG/wav2vec2-xlsr-300m-darija-3-augmentation-from-47"
32
  device = 0 if torch.cuda.is_available() else -1
33
  pipe = pipeline("automatic-speech-recognition", model=MODEL_NAME, device=device)
34
 
 
28
  )
29
 
30
  # Load the pipeline with device set to GPU if available
31
+ MODEL_NAME = "STT-Darija-ORG/wav2vec2-xlsr-300m-darija-augmented"
32
  device = 0 if torch.cuda.is_available() else -1
33
  pipe = pipeline("automatic-speech-recognition", model=MODEL_NAME, device=device)
34