Update inference.py

inference.py

@@ -12,6 +12,7 @@ logger = transformers.logging.get_logger('liveinfer')
 class LiveInfer:
     def __init__(self, ) -> None:
         args = parse_args()
+        args.resume_from_checkpoint = 'checkpoints/live1+_aug_2e'
         self.model, self.tokenizer = build_model_and_tokenizer(is_training=False, set_vision_inside=True, **asdict(args))
         self.model.to('cuda')