Update app.py

app.py

@@ -63,14 +63,15 @@ def vc_fn(sid, audio_record, audio_upload, tran, pndm_speedup=20):
     f0_pred[f0_pred==0]=np.nan#f0 pe predicted
     f0_gen[f0_gen==0]=np.nan#f0 generated
     fig=plt.figure(figsize=[15,5])
-    plt.plot(np.arange(0,len(f0_tst)),f0_tst,color='black')
-    plt.plot(np.arange(0,len(f0_pred)),f0_pred,color='orange')
-    plt.plot(np.arange(0,len(f0_gen)),f0_gen,color='red')
-    plt.axhline(librosa.note_to_hz('C4'),ls=":",c="blue")
-    plt.axhline(librosa.note_to_hz('G4'),ls=":",c="green")
-    plt.axhline(librosa.note_to_hz('C5'),ls=":",c="orange")
-    plt.axhline(librosa.note_to_hz('F#5'),ls=":",c="red")
-    #plt.axhline(librosa.note_to_hz('A#5'),ls=":",c="black")
+    plt.plot(np.arange(0,len(f0_tst)),f0_tst,color='black',label="f0_tst")
+    plt.plot(np.arange(0,len(f0_pred)),f0_pred,color='orange',label="f0_pred")
+    plt.plot(np.arange(0,len(f0_gen)),f0_gen,color='red',label="f0_gen")
+    plt.axhline(librosa.note_to_hz('C4'),ls=":",c="blue",label="C4")
+    plt.axhline(librosa.note_to_hz('G4'),ls=":",c="green",label="G4")
+    plt.axhline(librosa.note_to_hz('C5'),ls=":",c="orange",label="C5")
+    plt.axhline(librosa.note_to_hz('F#5'),ls=":",c="red",label="F#5")
+    #plt.axhline(librosa.note_to_hz('A#5'),ls=":",c="black",label="分割线")
+    plt.legend()
     plt.savefig('./temp.png')
 
     return "Success", (sr, audio), gr.Image.update("temp.png") # hparams['audio_sample_rate']
@@ -99,8 +100,7 @@ with app:
                         无用信息
                         """)
             out_message = gr.Textbox(label="Output")
-            gr.Markdown(value="""f0曲线可以直观的显示跑调情况，蓝色为输入音高，橙色为合成音频的音高
-                        若**只看见橙色**，说明蓝色曲线被覆盖，转换效果较好
+            gr.Markdown(value="""f0曲线可以直观的显示跑调情况：
                         """)
             f0_image = gr.Image(label="f0曲线")
         vc_submit.click(vc_fn, [speaker_id, record_input, upload_input, vc_transform, vc_speedup],