kotoba_whisper

Runtime error

aka7774 commited on Apr 30

Commit

e68b7ad

•

1 Parent(s): 7935c8d

Update fn.py

Files changed (1) hide show

fn.py CHANGED Viewed

@@ -42,7 +42,7 @@ def speech_to_text(audio_file, _model_size = None):
     # run inference
     generate_kwargs = {}
     if initial_prompt:
-        generate_kwargs['prompt_ids'] = pipe.tokenizer.get_prompt_ids(prompt, return_tensors="pt").to(device)
     result = pipe(audio_file, generate_kwargs=generate_kwargs)
     try:

     # run inference
     generate_kwargs = {}
     if initial_prompt:
+        generate_kwargs['prompt_ids'] = pipe.tokenizer.get_prompt_ids(initial_prompt, return_tensors="pt").to(device)
     result = pipe(audio_file, generate_kwargs=generate_kwargs)
     try: