TobDeBer commited on
Commit
8fdee99
·
1 Parent(s): 5c07d56
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -13,6 +13,7 @@ import os
13
  import re
14
 
15
  import pandas as pd
 
16
 
17
  import importlib
18
  modeling_MERT = importlib.import_module("MERT-v1-95M.modeling_MERT")
@@ -169,11 +170,11 @@ def model_inference(inputs):
169
  return df
170
 
171
  def convert_audio(inputs):
172
- return model_inference(inputs)
 
173
 
174
  def build_audio_flow(title, description, article):
175
- # Define file upload element
176
- audio_file = gr.File(label="Select Audio File (WAV, FLAC)")
177
 
178
  demo = gr.Interface(
179
  fn=convert_audio,
 
13
  import re
14
 
15
  import pandas as pd
16
+ import librosa
17
 
18
  import importlib
19
  modeling_MERT = importlib.import_module("MERT-v1-95M.modeling_MERT")
 
170
  return df
171
 
172
  def convert_audio(inputs):
173
+ audio_data, sample_rate = librosa.load(inputs, sr=None)
174
+ return model_inference(audio_data)
175
 
176
  def build_audio_flow(title, description, article):
177
+ audio_file = gr.File(label="Select Audio File")
 
178
 
179
  demo = gr.Interface(
180
  fn=convert_audio,