Update app.py

app.py

@@ -13,8 +13,6 @@ import tqdm
 from midi_synthesizer import synthesis
 import TMIDIX
 
-from math import ceil
-
 in_space = os.getenv("SYSTEM") == "spaces"
       
 #=================================================================================================
@@ -48,13 +46,7 @@ def GenerateMIDI():
     
             logits = torch.FloatTensor(session.run(None, {'input': [torch_in]})[0])[:, -1]
 
-            thres = 0.9
-            k = ceil((1 - thres) * logits.shape[-1])
-            val, ind = torch.topk(logits, k)
-            probs = torch.full_like(logits, float('-inf'))
-            probs.scatter_(1, ind, val)
-            
-            probs = F.softmax(probs / temperature, dim=-1)
+            probs = F.softmax(logits / temperature, dim=-1)
     
             sample = torch.multinomial(probs, 1)