Update app.py

app.py

@@ -404,6 +404,9 @@ if __name__ == "__main__":
         input_number_prime_tokens = gr.Slider(0, 1024, value=512, step=16, label="Number of prime tokens")
         input_number_gen_tokens = gr.Slider(0, 2048, value=512, step=16, label="Number of tokens to generate")
 
+        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")
+        
         run_btn = gr.Button("generate", variant="primary")
 
         gr.Markdown("## Generation results")
@@ -419,20 +422,24 @@ if __name__ == "__main__":
                                                        input_match_original_pitches_counts,
                                                        input_freestyle_continuation,
                                                        input_number_prime_tokens,
-                                                       input_number_gen_tokens
+                                                       input_number_gen_tokens,
+                                                       input_model_temperature,
+                                                       input_model_top_k
                                                       ],
                                   [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot])
 
         gr.Examples(
-            [["Sharing The Night Together.kar", 256, True, True, False, 512, 1024],
-             ["Hotel California.mid", 256, True, True, False, 512, 1024],
+            [["Sharing The Night Together.kar", 256, True, True, False, 512, 1024, 0.9, 10],
+             ["Hotel California.mid", 256, True, True, False, 512, 1024, 0.9, 10],
             ],
             [input_number_prime_chords, 
              input_use_original_durations,
              input_match_original_pitches_counts,
              input_freestyle_continuation,
              input_number_prime_tokens,
-             input_number_gen_tokens
+             input_number_gen_tokens,
+             input_model_temperature,
+             input_model_top_k
             ],
             [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot],
             Generate_Rock_Song,