Update app.py

app.py

@@ -6,10 +6,9 @@ import torch
 import torch.nn.functional as F
 
 import gradio as gr
-import numpy as np
+
 import onnxruntime as rt
 import tqdm
-import json
 
 from midi_synthesizer import synthesis
 import TMIDIX
@@ -18,9 +17,6 @@ in_space = os.getenv("SYSTEM") == "spaces"
       
 #=================================================================================================
 
-def create_msg(name, data):
-    return {"name": name, "data": data}
-
 def GenerateMIDI():
 
     start_tokens = [3087, 3073+1, 3075+1]
@@ -45,12 +41,14 @@ def GenerateMIDI():
         try:
     
             x = out[:, -max_seq_len:]
-            
+    
             torch_in = x.tolist()[0]
     
             logits = torch.FloatTensor(session.run(None, {'input': [torch_in]})[0])[:, -1]
             
-            probs = F.softmax(logits / temperature, dim=-1)
+            filtered_logits = logits
+    
+            probs = F.softmax(filtered_logits / temperature, dim=-1)
     
             sample = torch.multinomial(probs, 1)