app

app.py

@@ -39,7 +39,7 @@ import tempfile
 from PIL import Image
 import glob
 from src.data import DemoData
-from src.models import LINO_UniPS
+from src.models import LiNo_UniPS
 from torch.utils.data import DataLoader
 import pytorch_lightning as pl
 import spaces
@@ -378,7 +378,7 @@ if __name__ == "__main__":
 
     hi3dgen_pipeline = Hi3DGenPipeline.from_pretrained("weights/trellis-normal-v0-1")
     hi3dgen_pipeline.cuda()
-    lino = LINO_UniPS()
+    lino = LiNo_UniPS()
     lino.from_pretrained("weights/lino/lino.pth")
   
     demo.launch(share=False, server_name="0.0.0.0")