Update app.py

app.py

@@ -16,6 +16,8 @@ import MIDI
 from midi_synthesizer import synthesis
 from midi_tokenizer import MIDITokenizer
 
+in_space = os.getenv("SYSTEM") == "spaces"
+
 def softmax(x, axis):
     x_max = np.amax(x, axis=axis, keepdims=True)
     exp_x_shifted = np.exp(x - x_max)
@@ -58,7 +60,7 @@ def generate(prompt=None, max_len=512, temp=1.0, top_p=0.98, top_k=20,
         input_tensor = prompt
     input_tensor = input_tensor[None, :, :]
     cur_len = input_tensor.shape[1]
-    bar = tqdm.tqdm(desc="generating", total=max_len - cur_len)
+    bar = tqdm.tqdm(desc="generating", total=max_len - cur_len, disable=in_space)
     with bar:
         while cur_len < max_len:
             end = False
@@ -204,7 +206,7 @@ if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument("--share", action="store_true", default=False, help="share gradio app")
     parser.add_argument("--port", type=int, default=7860, help="gradio server port")
-    parser.add_argument("--max-gen", type=int, default=256, help="max")
+    parser.add_argument("--max-gen", type=int, default=512, help="max")
     opt = parser.parse_args()
     soundfont_path = hf_hub_download(repo_id="skytnt/midi-model", filename="soundfont.sf2")
     model_base_path = hf_hub_download(repo_id="skytnt/midi-model", filename="onnx/model_base.onnx")