Update app.py

app.py

@@ -128,7 +128,9 @@ print('=' * 70)
 def Generate_POP_Section(input_comp_section, 
                          input_mode_time, 
                          input_mode_dur,
-                         input_mode_ptc
+                         input_mode_ptc,
+                         input_model_temp,
+                         input_model_top_p
                          ):
     
     print('=' * 70)
@@ -142,6 +144,8 @@ def Generate_POP_Section(input_comp_section,
     print('Mode time:', input_mode_time)
     print('Mode duration:', input_mode_dur)
     print('Mode pitch:', input_mode_ptc)
+    print('Model temperature:', input_model_temp)
+    print('Model top p:', input_model_top_p)
     print('=' * 70)
 
     #===============================================================================
@@ -174,9 +178,9 @@ def Generate_POP_Section(input_comp_section,
         
         out = model.generate(x,
                              512-len(seq),
-                             temperature=0.9,
+                             temperature=input_model_temp,
                              filter_logits_fn=top_p,
-                             filter_kwargs={'thres': 0.96},
+                             filter_kwargs={'thres': input_model_top_p},
                              eos_token=917,
                              return_prime=True,
                              verbose=True)
@@ -202,7 +206,7 @@ def Generate_POP_Section(input_comp_section,
     comp_summary += 'Generated section: ' + str(comp_section) + '\n'
     comp_summary += 'Generated mode time: ' + str(comp_mode_time) + '\n'
     comp_summary += 'Generated mode duration: ' + str(comp_mode_dur) + '\n'
-    comp_summary += 'Generated mode pitch :' + str(comp_mode_ptc)
+    comp_summary += 'Generated mode pitch: ' + str(comp_mode_ptc)
     
     print('Sample INTs', song[:5])
     print('=' * 70)
@@ -273,7 +277,7 @@ def Generate_POP_Section(input_comp_section,
 
     print('Output MIDI file name:', output_midi)
     print('Output MIDI title:', output_midi_title)
-    print('Output MIDI summary', comp_summary)
+    print('Output MIDI summary:', comp_summary)
     print('=' * 70) 
     
 
@@ -318,6 +322,8 @@ if __name__ == "__main__":
         input_mode_time = gr.Slider(0, 127, value=0, step=1, label="Composition mode time")
         input_mode_dur = gr.Slider(0, 127, value=0, step=1, label="Composition mode dur")
         input_mode_ptc = gr.Slider(0, 127, value=0, step=1, label="Composition mode pitch")
+        input_model_temp = gr.Slider(0.1, 1, value=0.9, step=0.01, label="Model temperature")
+        input_model_top_p = gr.Slider(0.1, 1, value=0.96, step=0.01, label="Model sampling top p value")
         
         run_btn = gr.Button("Generate", variant="primary")
         
@@ -332,7 +338,9 @@ if __name__ == "__main__":
         run_event = run_btn.click(Generate_POP_Section, [input_comp_section, 
                                                         input_mode_time, 
                                                         input_mode_dur,
-                                                        input_mode_ptc
+                                                        input_mode_ptc,
+                                                        input_model_temp,
+                                                        input_model_top_p
                                                         ],
                                                         [output_midi_title,
                                                          output_midi_summary,
@@ -341,12 +349,16 @@ if __name__ == "__main__":
                                                          output_plot]
                                                         )
 
-        gr.Examples([["intro", 10, 15, 60],
+        gr.Examples([["intro", 10, 15, 72, 0.9, 0.96],
+                     ["chorus", 10, 15, 72, 0.9, 0.96],
+                     ["bridge", 10, 15, 72, 0.9, 0.96]
                     ],
                     [input_comp_section, 
                     input_mode_time, 
                     input_mode_dur,
-                    input_mode_ptc
+                    input_mode_ptc,
+                    input_model_temp,
+                    input_model_top_p
                     ],
                     [output_midi_title,
                     output_midi_summary,