asigalov61 commited on
Commit
eb9280b
·
verified ·
1 Parent(s): cee38fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -0
app.py CHANGED
@@ -264,6 +264,34 @@ def Generate_Accompaniment(input_midi,
264
  model.to(DEVICE)
265
  model.eval()
266
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
267
  #==================================================================
268
 
269
  print('=' * 70)
 
264
  model.to(DEVICE)
265
  model.eval()
266
 
267
+ #==================================================================
268
+
269
+ start_score_seq = [1792] + score + [1793]
270
+
271
+ #==================================================================
272
+
273
+ if generation_type == 'Guided':
274
+
275
+ input_seq = []
276
+
277
+ input_seq.extend(start_score_seq)
278
+ input_seq.extend(score_list[0][0])
279
+
280
+ for i in tqdm.tqdm(range(len(score_list)-1)):
281
+
282
+ input_seq.extend(score_list[i][1])
283
+
284
+ block_seq = generate_block_seq(input_seq, score_list[i+1][0][0], temperature=model_temperature)
285
+
286
+ input_seq.extend(block_seq)
287
+
288
+ else:
289
+ input_seq = generate_full_seq(start_score_seq, temperature=model_temperature)
290
+
291
+ #==================================================================
292
+
293
+ final_song = input_seq[len(start_score_seq):]
294
+
295
  #==================================================================
296
 
297
  print('=' * 70)