clr commited on
Commit
9f4e47e
·
1 Parent(s): c793959

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,14 +40,14 @@ def recc(a_f):
40
  input_values = processor(wav,sampling_rate=16000).input_values[0]
41
  input_values = torch.tensor(input_values, device=device).unsqueeze(0)
42
  logits = model(input_values).logits
43
- pred_ids = torch.argmax(logits, dim=-1)[0]
44
  return pred_ids
45
  #xcp = processor.decode(pred_ids)
46
  #return xcp
47
 
48
  def dec(pids):
49
  with torch.inference_mode():
50
- xcp = processor.decode(pred_ids)
51
  return xcp
52
 
53
  bl = gr.Blocks()
 
40
  input_values = processor(wav,sampling_rate=16000).input_values[0]
41
  input_values = torch.tensor(input_values, device=device).unsqueeze(0)
42
  logits = model(input_values).logits
43
+ pred_ids = torch.argmax(logits, dim=-1)[0] #s[0].cpu().detach()
44
  return pred_ids
45
  #xcp = processor.decode(pred_ids)
46
  #return xcp
47
 
48
  def dec(pids):
49
  with torch.inference_mode():
50
+ xcp = processor.decode(pids)
51
  return xcp
52
 
53
  bl = gr.Blocks()