Update app.py

app.py

@@ -147,38 +147,39 @@ def Generate_POP_Section(input_comp_section,
     #===============================================================================
 
     print('Generating...')
-
-    # seq = [916] + [part_tok+512, mode_time(score)+532, mode_dur(score)+660, mode_pitch(score)+788]
-    
-    part = 'chorus'
-    mode_time = 10
-    mode_dur = 10
-    mode_pitch = 66
-    
-    #seq = [916] + [parts_dict.index(part)+512, mode_time+532, mode_dur+660, mode_pitch+788]
     
-    #seq = random.choice(train_data)[:128]
+    if input_comp_section == 'random':
+        seq = [916]
+
+    else:
+        seq = [916, sections_dict.index(input_comp_section)+512]
+
+        input_seq = [input_mode_time, input_mode_dur, input_mode_ptc]
+        input_seq_toks = [input_mode_time+532, input_mode_dur+660, input_mode_ptc+788]
     
+        input_seq = input_seq_toks[:input_seq.index(0)]
+
+        seq += input_seq
+
     x = torch.LongTensor(seq).cuda()
     
     with ctx:
         out = model.generate(x,
-                             512,
+                             512-len(seq),
                              temperature=0.9,
-                             #filter_logits_fn=top_p,
-                             #filter_kwargs={'thres': 0.96},
+                             filter_logits_fn=top_p,
+                             filter_kwargs={'thres': 0.96},
                              eos_token=917,
                              return_prime=True,
                              verbose=True)
 
-    y = out.tolist()[0]
-
-    song.extend(y)
+    song = out.tolist()[0]
 
     print('Done!')
     print('=' * 70)
         
     #===============================================================================
+    
     print('Rendering results...')
     
     print('=' * 70)