Update model_helper.py

model_helper.py

@@ -1,4 +1,6 @@
 # @title Model helper
+import spaces # for zero-GPU
+
 import os
 from collections import Counter
 import argparse
@@ -17,9 +19,9 @@ from utils.event2note import merge_zipped_note_events_and_ties_to_notes
 from utils.utils import write_model_output_as_midi, write_err_cnt_as_json
 from model.ymt3 import YourMT3
 
-# import spaces # for zero-GPU
 
-# @spaces.GPU
+
+@spaces.GPU
 def load_model_checkpoint(args=None):
     parser = argparse.ArgumentParser(description="YourMT3")
     # General
@@ -120,7 +122,7 @@ def load_model_checkpoint(args=None):
     model.load_state_dict(new_state_dict, strict=False)
     return model.eval()
 
-# @spaces.GPU
+@spaces.GPU
 def transcribe(model, audio_info):
     t = Timer()