Update app.py

app.py

@@ -22,9 +22,9 @@ def restore(audio, original_transcript):
     print(error)
     data, samplerate = librosa.load(path, sr=16_000)
 
-    text, _ = restorer(data, original_transcript, samplerate, num_beams=1)
+    text, log_probs = restorer(data, original_transcript, samplerate, num_beams=1)
 
-    return text
+    return text, log_probs
 
 
 gr.Interface(
@@ -36,6 +36,7 @@ gr.Interface(
     ],
     outputs=[
         gr.outputs.Textbox(label='Restored text'),
+        gr.outputs.Number(label='Log probability')
     ],
     examples=[
         ["./common_voice_en_18301577.mp3", "do not cross the yellow light"],