Update app.py

app.py

@@ -16,9 +16,9 @@ labels = learn.dls.vocab
 def predict(Record, Upload):
     if Upload: path = Upload
     else: path = Record
-    pred,pred_idx,probs = learn.predict(str(path))
+    spec,pred,pred_idx,probs = learn.predict(str(path), with_input=True)
 
-    return {labels[i]: float(probs[i]) for i in range(len(labels))}
+    return [{labels[i]: float(probs[i]) for i in range(len(labels))},spec]
 
 
 title = "Japanese Pitch Accent Pattern Detector"
@@ -33,4 +33,4 @@ examples = [['代わる.mp3'],['大丈夫な.mp3'],['熱くない.mp3']]
 
 enable_queue=True
 
-gr.Interface(fn=predict,inputs=[gr.inputs.Audio(source='microphone', type='filepath', optional=True), gr.inputs.Audio(source='upload', type='filepath', optional=True)] ,outputs=gr.outputs.Label(num_top_classes=3),title=title,description=description,article=article,examples=examples).launch(debug=True,share=True,enable_queue=enable_queue)
+gr.Interface(fn=predict,inputs=[gr.inputs.Audio(source='microphone', type='filepath', optional=True), gr.inputs.Audio(source='upload', type='filepath', optional=True)], outputs=  [gr.outputs.Label(num_top_classes=3) 'image'], title=title,description=description,article=article,examples=examples).launch(debug=True,share=True,enable_queue=enable_queue)