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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -41,10 +41,14 @@ def recc(a_f):
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
 
49
  bl = gr.Blocks()
50
  with bl:
@@ -54,6 +58,10 @@ with bl:
54
 
55
  text_button.click(recc, inputs=audio_file, outputs=text_output)
56
 
 
 
 
 
57
 
58
  bl.launch()
59
 
 
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()
54
  with bl:
 
58
 
59
  text_button.click(recc, inputs=audio_file, outputs=text_output)
60
 
61
+ text_button2 = gr.Button("Dec")
62
+ text_output2 = gr.Textbox()
63
+ text_button2.click(dec, inputs=text_output, outputs=text_output2)
64
+
65
 
66
  bl.launch()
67