asif00 commited on
Commit
fb6eab4
·
verified ·
1 Parent(s): 081c26b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -4,11 +4,10 @@ import gradio as gr
4
  import torch
5
 
6
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
7
- asr = pipeline(model="asif00/whisper-bangla").to(device=device)
8
- ser = pipeline("text2text-generation", model="asif00/mbart_bn_error_correction").to(
9
- device=device
10
- )
11
-
12
 
13
  @spaces.GPU
14
  def transcribe(audio):
 
4
  import torch
5
 
6
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
7
+ asr = pipeline(model="asif00/whisper-bangla")
8
+ asr.to(device=device)
9
+ ser = pipeline("text2text-generation", model="asif00/mbart_bn_error_correction")
10
+ ser.to(device=device)
 
11
 
12
  @spaces.GPU
13
  def transcribe(audio):