Spaces:
Configuration error
Configuration error
File size: 1,202 Bytes
e8eb745 f358539 badf837 e8eb745 6aff340 e8eb745 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
from datetime import datetime
import gradio as gr
import mgzip
import numpy as np
from os.path import join
import pickle
from zipfile import ZipFile
from _Generation import Generator
import tensorflow as tf
from tensorflow import keras
gen = Generator()
def main(artist):
if artist == 'Any':
artist = None
gen.generate_track_batch(artist)
filename = f'generation_{datetime.now().strftime("%Y_%m_%d %H_%M_%S")}.gp5'
gen.save_tracks(filename)
# create a ZipFile object
zipObj = ZipFile(filename.replace('.gp5', '.zip'), "w")
zipObj.write(filename)
zipObj.close()
return filename.replace('.gp5', '.zip')
with mgzip.open(join('data', 'track_data.pickle.gz'), 'rb') as file:
track_data = pickle.load(file)
inputs = gr.components.Radio(['Any'] + list(track_data.artist.unique()),
label='''Choose an artist for song initialization. Paper: "Rock Guitar Tablature Generation via Natural Language Processing", https://arxiv.org/abs/2301.05295''')
i = gr.Interface(fn = main, inputs = inputs,
outputs = gr.components.File(label='Generated Guitar Tabs. Download and Unzip to View:'))
i.launch() |