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