Update app.py

app.py

@@ -36,11 +36,11 @@ st.sidebar.markdown(ABOUT_TEXT)
 st.sidebar.markdown(CONTACT_TEXT)
 
 
-def inference(audio, num_epoch, tonic_name, chord_duration):
+def inference(audio, num_epoch, chord_duration):
     generator = Generator()
     if chord_duration == 0.0:
         chord_duration = None
-    output_midi_data = generator(audio, num_epoch=int(num_epoch), tonic=tonic_name, chord_duration=chord_duration)[0]
+    output_midi_data = generator(audio, num_epoch=int(num_epoch), chord_duration=chord_duration)[0]
     name = uuid.uuid4()
     output_midi_data.write(f'{name}.mid')
     fs = FluidSynth("font.sf2")
@@ -54,7 +54,7 @@ def inference(audio, num_epoch, tonic_name, chord_duration):
 st.title("Accompaniment Generator")
 
 st.markdown(
-    "App to generate accompaniment for MIDI music file with Evolutionary algorithm. Check out [project repository](https://github.com/AlekseyKorshuk/accompaniment-generator) and project limitations.")
+    "App to generate accompaniment for MIDI music file with Evolutionary algorithm. Check out [project repository](https://github.com/AlekseyKorshuk/accompaniment-generator).")
 
 article = "<p style='text-align: center'>" \
           "<a href='https://github.com/AlekseyKorshuk/accompaniment-generator' target='_blank'>Github Repo</a>" \
@@ -89,11 +89,6 @@ chord_duration = st.number_input("Custom chord duration is seconds (leave zero f
                                  format="%.4f"
                                  )
 
-tonic_name = st.selectbox(
-    'Select tonic:',
-    ["minor", "major"]
-)
-
 generate_image_button = st.button("Generate")
 
 if generate_image_button:
@@ -104,7 +99,7 @@ if generate_image_button:
             f.write(uploaded_file.getvalue())
     # print(uploaded_file.getvalue())
     with st.spinner(text=f"Generating, this may take some time..."):
-        before, after = inference(input_file, num_epoch, tonic_name, chord_duration)
+        before, after = inference(input_file, num_epoch, chord_duration)
     st.markdown("Before:")
     st.audio(before)
     st.markdown("After:")