Update app.py

app.py

@@ -27,7 +27,7 @@ def GenerateMIDI():
     return_prime=False
     progress=gr.Progress()
 
-    out = torch.LongTensor([start_tokens])
+    out = torch.FloatTensor([start_tokens])
 
     st = len(start_tokens)
 
@@ -35,11 +35,12 @@ def GenerateMIDI():
       print("Generating sequence of max length:", seq_len)
 
     progress(0, desc="Starting...")
-                
-    for i in progress.tqdm(range(seq_len)):
 
-        try:
-            with torch.no_grad():
+    with torch.no_grad():
+    
+        for i in progress.tqdm(range(seq_len)):
+    
+            try:
                 x = out[:, -max_seq_len:]
         
                 torch_in = x.tolist()[0]
@@ -51,10 +52,10 @@ def GenerateMIDI():
                 sample = torch.multinomial(probs, 1)
         
                 out = torch.cat((out, sample), dim=-1)
-
-        except Exception as e:
-            print('Error', e)
-            break
+    
+            except Exception as e:
+                print('Error', e)
+                break
             
     if return_prime:
       melody_chords_f = out[:, :]