Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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": "
|
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)
|