update

app.py

@@ -5,12 +5,10 @@ import glob
 import torch
 import random
 from tempfile import NamedTemporaryFile
-from infer_api import InferAPI
 from PIL import Image
 import os
 import subprocess
 
-
 def install_cuda_toolkit():
     # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
     CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
@@ -30,6 +28,9 @@ def install_cuda_toolkit():
 
 install_cuda_toolkit()
 
+
+from infer_api import InferAPI
+
 config_canocalize = {
     'config_path': './configs/canonicalization-infer.yaml',
 }