Update pix2struct/inference.py

pix2struct/inference.py

@@ -3,7 +3,6 @@ import dataclasses
 import json
 from typing import Generator, Any
 
-from accelerate import Accelerator
 from transformers import T5TokenizerFast
 import numpy as np
 import torch
@@ -176,6 +175,7 @@ def main():
     args.add_argument('--queries', type=str, required=True)
     args = args.parse_args()
 
+    from accelerate import Accelerator
     accelerator = Accelerator()
     model = Pix2StructModel.load(args.model)
     model = accelerator.prepare(model)