Update app.py

app.py

@@ -291,33 +291,37 @@ class InferenceModel(object):
 			tokens = tokens[:np.argmax(tokens == vocabularies.DECODED_EOS_ID)]
 		return tokens
 
-
 inference_model = InferenceModel('/home/user/app/checkpoints/mt3/', 'mt3')
 
-
 def inference(audio):
-	filename = os.path.basename(audio)  # 获取输入文件的文件名
+	filename = os.path.basename(audio)
 	print(f"[{current_time()}] 运行：输入文件: {filename}")
 	with open(audio, 'rb') as fd:
 		contents = fd.read()
-	audio = upload_audio(contents,sample_rate=16000)
-	est_ns = inference_model(audio)
+	audio_data = upload_audio(contents, sample_rate=16000)
+	est_ns = inference_model(audio_data)
 	note_seq.sequence_proto_to_midi_file(est_ns, './transcribed.mid')
 	return './transcribed.mid'
 
-title = "MT3"
-description = "MT3：多任务多音轨音乐转录的 Gradio 演示。要使用它，只需上传音频文件，或点击示例以查看效果。更多信息请参阅下面的链接。"
-
-article = "<p style='text-align: center'>出错了？试试把文件转换为WAV后再上传吧~</p><p style='text-align: center'><a href='https://arxiv.org/abs/2111.03017' target='_blank'>MT3: 多任务多音轨音乐转录</a> | <a href='https://github.com/hmjz100/mt3' target='_blank'>Github 仓库</a></p>"
-
-examples=[['canon.flac'], ['download.wav']]
-
-gr.Interface(
-	inference,
-	gr.Audio(type="filepath", label="输入"),
-	outputs=gr.File(label="输出"),
-	title=title,
-	description=description,
-	article=article,
-	examples=examples
-).launch()
+with gr.Blocks(title="MT3", theme=gr.themes.Soft()) as demo:
+	gr.HTML("<p>MT3：多任务多音轨音乐转录的 Gradio 演示。要使用它，只需上传音频文件，或点击示例以查看效果。更多信息请参阅下面的链接。</p>")
+
+	with gr.Row():
+		with gr.Column():
+			audio_input = gr.Audio(type="filepath", label="输入", interactive=True)
+			gr.Examples(examples=[['canon.flac'], ['download.wav']], label="示例", inputs=audio_input)
+		midi_output = gr.File(label="输出")
+
+	submit_btn = gr.Button("开始", variant="primary")
+	submit_btn.click(
+		fn=inference,
+		inputs=audio_input,
+		outputs=midi_output
+	)
+	
+	gr.HTML('''<div class="footer">
+		<p style='text-align: center'>出错了？试试把文件转换为WAV后再上传吧~</p>
+		<p style='text-align: center'><a href='https://arxiv.org/abs/2111.03017' target='_blank'>MT3: 多任务多音轨音乐转录</a> | <a href='https://github.com/hmjz100/mt3' target='_blank'>Github 仓库</a></p>
+	</div>''')
+
+demo.launch(server_name="0.0.0.0", share=True)