Update app.py

app.py

@@ -3,6 +3,10 @@ import glob
 import json
 import os.path
 
+import time
+import datetime
+from pytz import timezone
+
 import torch
 import torch.nn.functional as F
 
@@ -16,13 +20,8 @@ import TMIDIX
 
 import matplotlib.pyplot as plt
 
-import time
-import datetime
-from pytz import timezone
-
 in_space = os.getenv("SYSTEM") == "spaces"
 
-
 # =================================================================================================
 
 @torch.no_grad()
@@ -85,13 +84,15 @@ def GenerateMIDI(num_tok, idrums, iinstr):
     yield output, None, None, [create_msg("visualizer_clear", None)]
 
     outy = start_tokens
+    
     ctime = 0
     dur = 0
-    vel = 0
+    vel = 90
     pitch = 0
     channel = 0
 
     for i in range(max(1, min(512, num_tok))):
+        
         inp = torch.LongTensor([outy]).cpu()
 
         out = model.module.generate(inp,
@@ -102,7 +103,8 @@ def GenerateMIDI(num_tok, idrums, iinstr):
 
         out0 = out[0].tolist()
         outy.extend(out0)
-        ss1 = int(out0[0])
+        
+        ss1 = out0[0]
 
         if 0 < ss1 < 256:
             ctime += ss1 * 8