Update app.py

app.py

@@ -35,7 +35,7 @@ def Generate_Rock_Song(input_midi,
                        input_number_gen_tokens,
                        input_num_memory_tokens,
                        input_model_temperature,
-                       input_model_top_k
+                       input_model_top_p
                       ):
 
     #===============================================================================
@@ -211,8 +211,8 @@ def Generate_Rock_Song(input_midi,
         with ctx:
           out = model.generate(x,
                               num_gen_tokens,
-                              filter_logits_fn=top_k,
-                              filter_kwargs={'k': input_model_top_k},
+                              filter_logits_fn=top_p,
+                              filter_kwargs={'thres': input_model_top_p},
                               temperature=input_model_temperature,
                               return_prime=True,
                               verbose=True)
@@ -246,8 +246,8 @@ def Generate_Rock_Song(input_midi,
               out = model.generate(x,
                                   1,
                                   filter_logits_fn=top_k,
-                                  filter_kwargs={'k': input_model_top_k},
-                                  temperature=input_model_temperature,
+                                  filter_logits_fn=top_p,
+                                  filter_kwargs={'thres': input_model_top_p},
                                   return_prime=False,
                                   verbose=False)
             
@@ -445,7 +445,7 @@ if __name__ == "__main__":
 
         input_num_memory_tokens = gr.Slider(1024, 4096, value=2048, step=16, label="Number of memory tokens")
         input_model_temperature = gr.Slider(0.1, 1.0, value=0.9, step=0.05, label="Model temperature")
-        input_model_top_k = gr.Slider(1, 50, value=10, step=1, label="Model sampling top k value")
+        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")
 
@@ -467,16 +467,16 @@ if __name__ == "__main__":
                                                        input_number_gen_tokens,
                                                        input_num_memory_tokens,
                                                        input_model_temperature,
-                                                       input_model_top_k,
+                                                       input_model_top_p,
                                                       ],
                                   [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot])
 
         gr.Examples(
-            [["Rock Violin.mid", "Controlled", 0, 512, True, True, 512, 1024, 2048, 0.9, 10],
-             ["Come To My Window.mid", "Controlled", 128, 256, False, False, 512, 1024, 2048, 0.9, 10],
-             ["Sharing The Night Together.kar", "Controlled", 128, 256, True, True, 512, 1024, 2048, 0.9, 10],
-             ["Hotel California.mid", "Controlled", 128, 256, True, True, 512, 1024, 2048, 0.9, 10],
-             ["Nothing Else Matters.kar", "Controlled", 128, 256, True, True, 512, 1024, 2048, 0.9, 10],
+            [["Rock Violin.mid", "Controlled", 0, 512, True, True, 512, 1024, 2048, 0.9, 0.96],
+             ["Come To My Window.mid", "Controlled", 128, 256, False, False, 512, 1024, 2048, 0.9, 0.96],
+             ["Sharing The Night Together.kar", "Controlled", 128, 256, True, True, 512, 1024, 2048, 0.9, 0.96],
+             ["Hotel California.mid", "Controlled", 128, 256, True, True, 512, 1024, 2048, 0.9, 0.96],
+             ["Nothing Else Matters.kar", "Controlled", 128, 256, True, True, 512, 1024, 2048, 0.9, 0.96],
             ],
             [input_midi,
              input_gen_type,
@@ -488,7 +488,7 @@ if __name__ == "__main__":
              input_number_gen_tokens,
              input_num_memory_tokens,
              input_model_temperature,
-             input_model_top_k,
+             input_model_top_p,
             ],
             [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot],
             Generate_Rock_Song,