juancopi81 commited on
Commit
c970deb
1 Parent(s): ef98e8e

Add upload_audio to utils file

Browse files
Files changed (3) hide show
  1. app.py +1 -11
  2. inferencemodel.py +0 -2
  3. utils.py +7 -0
app.py CHANGED
@@ -4,27 +4,17 @@ os.system("python3 -m pip install -e .")
4
 
5
  import gradio as gr
6
 
7
- import note_seq
8
-
9
- import nest_asyncio
10
- nest_asyncio.apply()
11
-
12
  from inferencemodel import InferenceModel
 
13
 
14
  SAMPLE_RATE = 16000
15
  SF2_PATH = 'SGM-v2.01-Sal-Guit-Bass-V1.3.sf2'
16
 
17
- def upload_audio(audio, sample_rate):
18
-
19
- return note_seq.audio_io.wav_data_to_samples_librosa(
20
- audio, sample_rate=sample_rate)
21
-
22
  # Start inference model
23
  inference_model = InferenceModel('/home/user/app/checkpoints/mt3/', 'mt3')
24
 
25
  def inference(audio):
26
  with open(audio, 'rb') as fd:
27
- print("acaasd", audio)
28
  contents = fd.read()
29
 
30
  audio = upload_audio(contents,sample_rate=16000)
 
4
 
5
  import gradio as gr
6
 
 
 
 
 
 
7
  from inferencemodel import InferenceModel
8
+ from utils import upload_audio
9
 
10
  SAMPLE_RATE = 16000
11
  SF2_PATH = 'SGM-v2.01-Sal-Guit-Bass-V1.3.sf2'
12
 
 
 
 
 
 
13
  # Start inference model
14
  inference_model = InferenceModel('/home/user/app/checkpoints/mt3/', 'mt3')
15
 
16
  def inference(audio):
17
  with open(audio, 'rb') as fd:
 
18
  contents = fd.read()
19
 
20
  audio = upload_audio(contents,sample_rate=16000)
inferencemodel.py CHANGED
@@ -1,5 +1,4 @@
1
  import functools
2
- import os
3
 
4
  import numpy as np
5
  import tensorflow.compat.v2 as tf
@@ -19,7 +18,6 @@ from mt3 import preprocessors
19
  from mt3 import spectrograms
20
  from mt3 import vocabularies
21
 
22
-
23
  import nest_asyncio
24
  nest_asyncio.apply()
25
 
 
1
  import functools
 
2
 
3
  import numpy as np
4
  import tensorflow.compat.v2 as tf
 
18
  from mt3 import spectrograms
19
  from mt3 import vocabularies
20
 
 
21
  import nest_asyncio
22
  nest_asyncio.apply()
23
 
utils.py ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+
2
+ import note_seq
3
+
4
+ def upload_audio(audio, sample_rate):
5
+
6
+ return note_seq.audio_io.wav_data_to_samples_librosa(
7
+ audio, sample_rate=sample_rate)