Update app.py

app.py

@@ -109,8 +109,10 @@ def empty_note_sequence(qpm=120.0, total_time=0.0):
     note_sequence.total_time = total_time
     return note_sequence
 
-def process(num1, num2):
-    created_text = f"""PIECE_START STYLE=JSFAKES GENRE=JSFAKES TRACK_START INST={num1} BAR_START NOTE_ON={num2}"""   
+def process(num1, num2, num3):
+    created_text = f"""PIECE_START STYLE=JSFAKES GENRE=JSFAKES TRACK_START INST={num1} BAR_START NOTE_ON={num2}"""
+    NOTE_LENGTH_16TH_120BPM = 0.25 * 60 / num3
+    BAR_LENGTH_120BPM = 4.0 * 60 / num3
     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])
@@ -149,10 +151,11 @@ with gr.Blocks() as demo:
         #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") 
+        number3 = gr.Slider(60, 140, value=120, label="BPM", step=5, info="Choose between 60 and 140")
         audio_button = gr.Button("generate audio")
     
     text_button.click(generation, inputs=text_input, outputs=text_output)
-    audio_button.click(process, inputs=[number1, number2], outputs=audio_output)
+    audio_button.click(process, inputs=[number1, number2, number3], outputs=audio_output)
 
 
 if __name__ == "__main__":