asahi417 commited on
Commit
5bd4675
1 Parent(s): 1f8a9e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ def format_time(start: Optional[float], end: Optional[float]):
61
 
62
  @spaces.GPU
63
  def get_prediction(inputs, prompt: Optional[str]):
64
- generate_kwargs = {"language": "japanese", "task": "transcribe"}
65
  if prompt:
66
  generate_kwargs['prompt_ids'] = pipe.tokenizer.get_prompt_ids(prompt, return_tensors='pt').to(device)
67
  prediction = pipe(inputs, return_timestamps=True, generate_kwargs=generate_kwargs)
 
61
 
62
  @spaces.GPU
63
  def get_prediction(inputs, prompt: Optional[str]):
64
+ generate_kwargs = {"language": "ja", "task": "transcribe"}
65
  if prompt:
66
  generate_kwargs['prompt_ids'] = pipe.tokenizer.get_prompt_ids(prompt, return_tensors='pt').to(device)
67
  prediction = pipe(inputs, return_timestamps=True, generate_kwargs=generate_kwargs)