MaximilianChen commited on
Commit
48a358e
1 Parent(s): 644d6d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -1,9 +1,15 @@
1
  from transformers import pipeline
2
- from transformers import WhisperProcessor, WhisperForConditionalGeneration
3
  import gradio as gr
 
4
 
 
5
 
6
- asr = pipeline(model="MaximilianChen/Casper")
 
 
 
 
 
7
 
8
  def transcribe_audio(mic=None, file=None):
9
  if mic is not None:
 
1
  from transformers import pipeline
 
2
  import gradio as gr
3
+ import torch
4
 
5
+ device = "cuda:0" if torch.cuda.is_available() else "cpu"
6
 
7
+ asr = pipeline(
8
+ "automatic-speech-recognition",
9
+ model="MaximilianChen/Casper",
10
+ chunk_length_s=30,
11
+ device=device,
12
+ )
13
 
14
  def transcribe_audio(mic=None, file=None):
15
  if mic is not None: