Update app.py

app.py

@@ -22,7 +22,7 @@ def GenerateMIDI():
     start_tokens = [3087, 3073+1, 3075+1]
     seq_len = 512
     max_seq_len = 2048
-    temperature = 0.9
+    temperature = 1.0
     verbose=False
     return_prime=False
     progress=gr.Progress()
@@ -46,9 +46,7 @@ def GenerateMIDI():
     
             logits = torch.FloatTensor(session.run(None, {'input': [torch_in]})[0])[:, -1]
             
-            filtered_logits = logits
-    
-            probs = F.softmax(filtered_logits / temperature, dim=-1)
+            probs = F.softmax(logits / temperature, dim=-1)
     
             sample = torch.multinomial(probs, 1)