Update app.py

app.py

@@ -281,6 +281,10 @@ def generate_music(prime,
         inputs.extend([0, ((8*8)+6)+256, ((128*129)+drum_pitch)+2304])
         
     torch.cuda.empty_cache()
+    model.cuda()
+    model.eval()
+
+    print('Generating...')
     
     inp = [inputs] * num_gen_batches
     
@@ -306,6 +310,9 @@ def generate_music(prime,
 
     else:
         output_batches = output
+
+    print('Done!')
+    print('=' * 70)
             
     return output_batches
     
@@ -398,6 +405,10 @@ def generate_callback_wrapper(input_midi,
                               model_sampling_top_p
                              ):
 
+    print('=' * 70)
+    print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
+    start_time = reqtime.time()
+    
     print('=' * 70)
     if input_midi is not None:
             fn = os.path.basename(input_midi.name)
@@ -422,6 +433,12 @@ def generate_callback_wrapper(input_midi,
                              )
     
     generated_batches.extend([sublist[2] for sublist in result])
+
+    print('=' * 70)
+    print('Req end time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
+    print('=' * 70)
+    print('Req execution time:', (reqtime.time() - start_time), 'sec')
+    print('*' * 70)    
     
     return tuple(item for sublist in result for item in sublist[:2])