Update app.py

app.py

@@ -109,9 +109,9 @@ def empty_note_sequence(qpm=120.0, total_time=0.0):
     note_sequence.total_time = total_time
     return note_sequence
 
-def process(text):
-    preset_text = created_sentence
-    input_ids = tokenizer.encode(preset_text, return_tensors="pt")
+def process(num1, num2):
+    created_text = f"""PIECE_START STYLE=JSFAKES GENRE=JSFAKES TRACK_START INST={num1} BAR_START NOTE_ON={num2}"""   
+    input_ids = tokenizer.encode(created_text, return_tensors="pt")
     generated_ids = model.generate(input_ids, max_length=500)
     generated_sequence = tokenizer.decode(generated_ids[0])
 
@@ -137,9 +137,6 @@ def identity(x, state):
     state += 1
     return x, state, state    
 
-def sentence_builder(quantity1, quantity2):
-    created_sentence = f"""PIECE_START STYLE=JSFAKES GENRE=JSFAKES TRACK_START INST={quantity1} BAR_START NOTE_ON={quantity2}"""
-
 with gr.Blocks() as demo:
     gr.Markdown("Midi Generation")
     with gr.Tab("Token generation"):
@@ -149,15 +146,13 @@ with gr.Blocks() as demo:
     with gr.Tab("Audio generation"):
         text_input = gr.Textbox()
         audio_output = gr.Audio()
+        #audio_button = gr.Button("generate audio")
+        number1 = gr.Slider(1, 100, value=25, label="Inst number", step=1, info="Choose between 1 and 100")
+        number2 = gr.Slider(1, 100, value=40, label="Note number", step=1, info="Choose between 1 and 100") 
         audio_button = gr.Button("generate audio")
-        sentence_builder,
-        [
-            gr.Slider(1, 100, value=25, label="Inst number", step=1, info="Choose between 1 and 100"),
-            gr.Slider(1, 100, value=40, label="Note number", step=1, info="Choose between 1 and 100") 
-        ]
     
     text_button.click(generation, inputs=text_input, outputs=text_output)
-    audio_button.click(process, inputs=text_input, outputs=audio_output)
+    audio_button.click(process, inputs=[number1, number2], outputs=audio_output)
 
 
 if __name__ == "__main__":