Update app.py

app.py

@@ -36,26 +36,24 @@ def GenerateMIDI():
 
     progress(0, desc="Starting...")
 
-    with torch.no_grad():
+    for i in progress.tqdm(range(seq_len)):
+
+        try:
+            x = out[:, -max_seq_len:]
     
-        for i in progress.tqdm(range(seq_len)):
+            torch_in = x.tolist()[0]
     
-            try:
-                x = out[:, -max_seq_len:]
-        
-                torch_in = x.tolist()[0]
-        
-                logits = torch.FloatTensor(session.run(None, {'input': [torch_in]})[0])[:, -1]
+            logits = torch.FloatTensor(session.run(None, {'input': [torch_in]})[0])[:, -1]
+
+            probs = F.softmax(logits / temperature, dim=-1)
     
-                probs = F.softmax(logits / temperature, dim=-1)
-        
-                sample = torch.multinomial(probs, 1)
-        
-                out = torch.cat((out, sample), dim=-1)
+            sample = torch.multinomial(probs, 1)
     
-            except Exception as e:
-                print('Error', e)
-                break
+            out = torch.cat((out, sample), dim=-1)
+
+        except Exception as e:
+            print('Error', e)
+            break
             
     if return_prime:
       melody_chords_f = out[:, :]