Update app.py
Browse files
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 |
-
|
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 |
|