Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -314,7 +314,7 @@ def tokens_to_MIDI(tokens, MIDI_name):
|
|
| 314 |
|
| 315 |
# =================================================================================================
|
| 316 |
|
| 317 |
-
|
| 318 |
def CompareMIDIs(input_src_midi, input_trg_midi, input_sampling_resolution, input_sampling_overlap):
|
| 319 |
|
| 320 |
print('=' * 70)
|
|
@@ -325,7 +325,7 @@ def CompareMIDIs(input_src_midi, input_trg_midi, input_sampling_resolution, inpu
|
|
| 325 |
|
| 326 |
SEQ_LEN = 8192 # Models seq len
|
| 327 |
PAD_IDX = 19463 # Models pad index
|
| 328 |
-
DEVICE = '
|
| 329 |
|
| 330 |
# instantiate the model
|
| 331 |
|
|
@@ -524,7 +524,7 @@ if __name__ == "__main__":
|
|
| 524 |
|
| 525 |
gr.Markdown("### Make sure that the MIDI has at least sampling resolution number of notes")
|
| 526 |
|
| 527 |
-
input_sampling_resolution = gr.Slider(50, 2000, value=
|
| 528 |
|
| 529 |
gr.Markdown("### Make sure that the sampling overlap value is less than sampling resolution value")
|
| 530 |
|
|
|
|
| 314 |
|
| 315 |
# =================================================================================================
|
| 316 |
|
| 317 |
+
@spaces.GPU
|
| 318 |
def CompareMIDIs(input_src_midi, input_trg_midi, input_sampling_resolution, input_sampling_overlap):
|
| 319 |
|
| 320 |
print('=' * 70)
|
|
|
|
| 325 |
|
| 326 |
SEQ_LEN = 8192 # Models seq len
|
| 327 |
PAD_IDX = 19463 # Models pad index
|
| 328 |
+
DEVICE = 'cuda' # 'cuda'
|
| 329 |
|
| 330 |
# instantiate the model
|
| 331 |
|
|
|
|
| 524 |
|
| 525 |
gr.Markdown("### Make sure that the MIDI has at least sampling resolution number of notes")
|
| 526 |
|
| 527 |
+
input_sampling_resolution = gr.Slider(50, 2000, value=50, step=10, label="Sampling resolution in notes")
|
| 528 |
|
| 529 |
gr.Markdown("### Make sure that the sampling overlap value is less than sampling resolution value")
|
| 530 |
|