disable tqdm

app.py

@@ -51,7 +51,7 @@ def generate(model: MIDIModel, prompt=None, batch_size=1, max_len=512, temp=1.0,
                             mode="constant", constant_values=tokenizer.pad_id)
         input_tensor = torch.from_numpy(prompt).to(dtype=torch.long, device=model.device)
     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)
     cache1 = DynamicCache()
     past_len = 0
     with bar:

app_onnx.py

@@ -107,7 +107,7 @@ def generate(model, prompt=None, batch_size=1, max_len=512, temp=1.0, top_p=0.98
                             mode="constant", constant_values=tokenizer.pad_id)
         input_tensor = prompt
     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)
     model0_inputs = {}
     model0_outputs = {}
     emb_size = 1024