Hugging Face's logo

Spaces:
kanyekuthi
/
kanyekuthi-AfriSpeech-whisper-tiny
Runtime error

App Files Community
1
kanyekuthi commited on
Commit
2b3994d
•
1 Parent(s): 73a19ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ model = AutoModelForSpeechSeq2Seq.from_pretrained(
54
  "openai/whisper-small", torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True, use_flash_attention_2=use_flash_attention_2
55
  )
56
  distilled_model = AutoModelForSpeechSeq2Seq.from_pretrained(
57
- "models/kanyekuthi/AfriSpeech-whisper-tiny", torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True, use_flash_attention_2=use_flash_attention_2
58
  )
59
 
60
  if not use_flash_attention_2:
 
54
  "openai/whisper-small", torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True, use_flash_attention_2=use_flash_attention_2
55
  )
56
  distilled_model = AutoModelForSpeechSeq2Seq.from_pretrained(
57
+ "distil-whisper/distil-small", torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True, use_flash_attention_2=use_flash_attention_2
58
  )
59
 
60
  if not use_flash_attention_2: