Spaces:

Achitha
/

simpletamil

Runtime error

Achitha commited on Jan 13, 2023

Commit

39c4239

1 Parent(s): 24dce6c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import pytube as pt
 from transformers import pipeline
 from huggingface_hub import model_info
-MODEL_NAME = "Achitha/whisper_small_ta" #this always needs to stay in line 8 :D sorry for the hackiness
 lang = "ta"
 device = 0 if torch.cuda.is_available() else "cpu"
@@ -16,7 +16,7 @@ pipe = pipeline(
     device=device,
 )
-pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language=lang, task="translate")
 def transcribe(microphone, file_upload):
     warn_output = ""

 from transformers import pipeline
 from huggingface_hub import model_info
+MODEL_NAME = "parambharat/whisper-small-ta" #this always needs to stay in line 8 :D sorry for the hackiness
 lang = "ta"
 device = 0 if torch.cuda.is_available() else "cpu"
     device=device,
 )
+pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language=lang, task="transcribe")
 def transcribe(microphone, file_upload):
     warn_output = ""