Hugging Face's logo

Spaces:
zeihers-mart
/
lab2
Runtime error

App Files Community
fabianzeiher commited on
Commit
45293a9
·
1 Parent(s): 73fe0d1

another attempt at forcing Swedish

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  from transformers import pipeline
2
  import gradio as gr
3
 
4
- pipe_fine = pipeline(model="zeihers-mart/whisper-small-swedish-basic", device_map="auto", language="swedish")
5
- pipe_raw = pipeline(model="openai/whisper-small", device_map="auto", language="swedish")
6
 
7
  def transcribe(audio):
8
  text_sv = pipe_fine(audio)["text"]
 
1
  from transformers import pipeline
2
  import gradio as gr
3
 
4
+ pipe_fine = pipeline(model="zeihers-mart/whisper-small-swedish-basic", device_map="auto", model_kwarks={"language":"swedish", "task":"transcribe"})
5
+ pipe_raw = pipeline(model="openai/whisper-small", device_map="auto", model_kwarks={"language":"swedish", "task":"transcribe"})
6
 
7
  def transcribe(audio):
8
  text_sv = pipe_fine(audio)["text"]