chore: add AutoImageProcessor

app.py

@@ -8,7 +8,7 @@ import mcubes
 import imageio
 from torchvision.utils import save_image
 from PIL import Image
-from transformers import AutoModel, AutoConfig
+from transformers import AutoModel, AutoConfig, AutoImageProcessor
 from rembg import remove, new_session
 from functools import partial
 from kiui.op import recenter
@@ -45,6 +45,7 @@ class LRMGeneratorWrapper:
     def __init__(self):
         self.config = AutoConfig.from_pretrained("yanranxiaoxi/image-upscale", trust_remote_code=True, token=os.environ.get('MODEL_ACCESS_TOKEN'))
         self.model = AutoModel.from_pretrained("yanranxiaoxi/image-upscale", trust_remote_code=True, token=os.environ.get('MODEL_ACCESS_TOKEN'))
+        self.image_processor = AutoImageProcessor.from_pretrained("yanranxiaoxi/image-upscale", trust_remote_code=True, token=os.environ.get('MODEL_ACCESS_TOKEN'), use_fast=True)
         self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         self.model.to(self.device)
         self.model.eval()