asigalov61 commited on
Commit
dae2bf7
1 Parent(s): 7194e86

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -66
app.py CHANGED
@@ -313,73 +313,7 @@ def InpaintPitches(input_midi, input_num_of_notes, input_patch_number):
313
 
314
  #==========================================================================
315
 
316
- nidx = 0
317
- first_inote = True
318
- fidx = 0
319
 
320
- number_of_prime_tokens = number_of_prime_notes * 3
321
-
322
- for i, m in enumerate(melody_chords):
323
-
324
- if 2304 <= melody_chords[i] < 18945:
325
-
326
- cpatch = (melody_chords[i]-2304) // 129
327
-
328
- if cpatch == inpaint_MIDI_patch:
329
- nidx += 1
330
- if first_inote:
331
- fidx += 1
332
-
333
- if first_inote and fidx == number_of_prime_notes:
334
- number_of_prime_tokens = i
335
- first_inote = False
336
-
337
- if nidx == input_num_of_notes:
338
- break
339
-
340
- nidx = i
341
-
342
- #==========================================================================
343
-
344
- out2 = []
345
-
346
- for m in melody_chords[:number_of_prime_tokens]:
347
- out2.append(m)
348
-
349
- for i in range(number_of_prime_tokens, len(melody_chords[:nidx])):
350
-
351
- cpatch = (melody_chords[i]-2304) // 129
352
-
353
- if 2304 <= melody_chords[i] < 18945 and (cpatch) == inpaint_MIDI_patch:
354
-
355
- samples = []
356
-
357
- for j in range(number_of_samples_per_inpainted_note):
358
-
359
- inp = torch.LongTensor(out2[-number_of_memory_tokens:]).cuda()
360
-
361
- with ctx:
362
- out1 = model.generate(inp,
363
- 1,
364
- temperature=temperature,
365
- return_prime=True,
366
- verbose=False)
367
-
368
- with torch.no_grad():
369
- test_loss, test_acc = model(out1)
370
-
371
- samples.append([out1.tolist()[0][-1], test_acc.tolist()])
372
-
373
- accs = [y[1] for y in samples]
374
- max_acc = max(accs)
375
- max_acc_sample = samples[accs.index(max_acc)][0]
376
-
377
- cpitch = (max_acc_sample-2304) % 129
378
-
379
- out2.extend([((cpatch * 129) + cpitch)+2304])
380
-
381
- else:
382
- out2.append(melody_chords[i])
383
 
384
  print('=' * 70)
385
  print('Done!')
 
313
 
314
  #==========================================================================
315
 
 
 
 
316
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
317
 
318
  print('=' * 70)
319
  print('Done!')