juancopi81 commited on
Commit
2a4494f
1 Parent(s): b487a20

Return audio and audio path

Browse files
Files changed (1) hide show
  1. app.py +11 -6
app.py CHANGED
@@ -53,11 +53,14 @@ def get_audio(url):
53
  def populate_metadata(link):
54
  yt = YouTube(link)
55
  audio = get_audio(link)
56
- return yt.thumbnail_url, yt.title, audio
57
 
58
- def inference(yt_audio):
59
-
60
- audio = upload_audio(yt_audio,sample_rate=SAMPLE_RATE)
 
 
 
61
 
62
  est_ns = inference_model(audio)
63
 
@@ -104,8 +107,9 @@ with demo:
104
  img = gr.Image(label="Thumbnail")
105
  with gr.Row():
106
  yt_audio = gr.Audio()
 
107
 
108
- link.change(fn=populate_metadata, inputs=link, outputs=[img, title, yt_audio])
109
 
110
  with gr.Row():
111
  btn = gr.Button("Transcribe music")
@@ -114,7 +118,8 @@ with demo:
114
  midi_file = gr.File()
115
  midi_audio = gr.Audio()
116
 
117
- btn.click(inference(yt_audio),
 
118
  outputs=[midi_file, midi_audio])
119
 
120
  gr.Markdown(article)
 
53
  def populate_metadata(link):
54
  yt = YouTube(link)
55
  audio = get_audio(link)
56
+ return yt.thumbnail_url, yt.title, audio, audio
57
 
58
+ def inference(yt_audio_path):
59
+
60
+ with open(yt_audio_path, 'rb') as fd:
61
+ contents = fd.read()
62
+
63
+ audio = upload_audio(contents,sample_rate=SAMPLE_RATE)
64
 
65
  est_ns = inference_model(audio)
66
 
 
107
  img = gr.Image(label="Thumbnail")
108
  with gr.Row():
109
  yt_audio = gr.Audio()
110
+ yt_audio_path = gr.Textbox(visible=False)
111
 
112
+ link.change(fn=populate_metadata, inputs=link, outputs=[img, title, yt_audio, yt_audio_path])
113
 
114
  with gr.Row():
115
  btn = gr.Button("Transcribe music")
 
118
  midi_file = gr.File()
119
  midi_audio = gr.Audio()
120
 
121
+ btn.click(inference,
122
+ inputs=yt_audio_path,
123
  outputs=[midi_file, midi_audio])
124
 
125
  gr.Markdown(article)