Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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)
|