Spaces:

amirza
/

draw_me_a_sheep_heb

Runtime error

Amir Zait commited on Aug 18, 2022

Commit

0d9345a

•

1 Parent(s): d8ec8f4

fixed bugs

Files changed (1) hide show

app.py CHANGED Viewed

@@ -58,7 +58,7 @@ def parse_transcription(wav_file):
     logits = asr_model(input_values).logits
     predicted_ids = torch.argmax(logits, dim=-1)
     transcription = asr_processor.decode(predicted_ids[0], skip_special_tokens=True)
-    translated = he_en_translator(trasncription)
     return translated
 output = gr.outputs.Textbox(label="TEXT")

     logits = asr_model(input_values).logits
     predicted_ids = torch.argmax(logits, dim=-1)
     transcription = asr_processor.decode(predicted_ids[0], skip_special_tokens=True)
+    translated = he_en_translator(transcription)
     return translated
 output = gr.outputs.Textbox(label="TEXT")