Harveenchadha
commited on
Commit
•
23ba586
1
Parent(s):
c043038
Update app.py
Browse files
app.py
CHANGED
@@ -45,7 +45,7 @@ def parse_transcription_with_lm(wav_file):
|
|
45 |
|
46 |
|
47 |
with torch.no_grad():
|
48 |
-
logits =
|
49 |
|
50 |
result = self.processor.batch_decode(logits.cpu().numpy())
|
51 |
text = result.text
|
|
|
45 |
|
46 |
|
47 |
with torch.no_grad():
|
48 |
+
logits = model(**input_values).logits
|
49 |
|
50 |
result = self.processor.batch_decode(logits.cpu().numpy())
|
51 |
text = result.text
|