Spaces:

sarulab-speech
/

CoCoCap-beta

Running on Zero

Wataru commited on Jan 24

Commit

a9075b6

•

1 Parent(s): 0871d92

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,10 +7,9 @@ transcribe = pipeline(
     model="ckpt_base/checkpoint-1308",
     tokenizer="ckpt_base",
     chunk_length_s=30,
-    device='cuda',
 )
 transcribe.model.config.forced_decoder_ids = transcribe.tokenizer.get_decoder_prompt_ids(language="ja", task="transcribe")
-transcribe.to('cuda')
 @spaces.GPU
 def main(audio_path):

     model="ckpt_base/checkpoint-1308",
     tokenizer="ckpt_base",
     chunk_length_s=30,
+    device=0,
 )
 transcribe.model.config.forced_decoder_ids = transcribe.tokenizer.get_decoder_prompt_ids(language="ja", task="transcribe")
 @spaces.GPU
 def main(audio_path):