Update app.py

app.py

@@ -36,9 +36,9 @@ st.sidebar.markdown(ABOUT_TEXT)
 st.sidebar.markdown(CONTACT_TEXT)
 
 
-def inference(audio, num_epoch):
+def inference(audio, num_epoch, tonic_name):
     generator = Generator()
-    output_midi_data = generator(audio, num_epoch=int(num_epoch))[0]
+    output_midi_data = generator(audio, num_epoch=int(num_epoch), tonic=tonic_name)[0]
     name = uuid.uuid4()
     output_midi_data.write(f'{name}.mid')
     fs = FluidSynth("font.sf2")
@@ -79,6 +79,11 @@ num_epoch = st.number_input("Number of epochs:",
                             value=10,
                             )
 
+tonic_name = st.selectbox(
+    'Select tonic:',
+    ["minor", "major"]
+)
+
 generate_image_button = st.button("Generate")
 
 if generate_image_button:
@@ -86,7 +91,7 @@ if generate_image_button:
     if uploaded_file is not None:
         input_file = uploaded_file.name
     with st.spinner(text=f"Generating, this may take some time..."):
-        before, after = inference(input_file, num_epoch)
+        before, after = inference(input_file, num_epoch, tonic_name)
     st.markdown("Before:")
     st.audio(before)
     st.markdown("After:")