Update inference.py

inference.py

@@ -6,11 +6,11 @@ from pathlib import Path
 
 import torch
 
-from .utilities import (create_folder, get_filename, RegressionPostProcessor, 
+from utilities import (create_folder, get_filename, RegressionPostProcessor, 
     write_events_to_midi)
-from .models import Regress_onset_offset_frame_velocity_CRNN, Note_pedal
-from .pytorch_utils import move_data_to_device, forward
-from . import config
+from models import Regress_onset_offset_frame_velocity_CRNN, Note_pedal
+from pytorch_utils import move_data_to_device, forward
+import config
 
 
 class PianoTranscription(object):
@@ -24,15 +24,9 @@ class PianoTranscription(object):
           segment_samples: int
           device: 'cuda' | 'cpu'
         """
-        if not checkpoint_path: 
-            checkpoint_path='{}/piano_transcription_inference_data/note_F1=0.9677_pedal_F1=0.9186.pth'.format(str(Path.home()))
-        print('Checkpoint path: {}'.format(checkpoint_path))
 
-        if not os.path.exists(checkpoint_path) or os.path.getsize(checkpoint_path) < 1.6e8:
-            create_folder(os.path.dirname(checkpoint_path))
-            print('Total size: ~165 MB')
-            zenodo_path = 'https://zenodo.org/record/4034264/files/CRNN_note_F1%3D0.9677_pedal_F1%3D0.9186.pth?download=1'
-            os.system('wget -O "{}" "{}"'.format(checkpoint_path, zenodo_path))
+        checkpoint_path="CRNN_note_F1%3D0.9677_pedal_F1%3D0.9186.pth"
+        print('Checkpoint path: {}'.format(checkpoint_path))
 
         print('Using {} for inference.'.format(device))