Move weights to huggingface models

.gitignore

@@ -8,4 +8,5 @@ flagged
 Synthetic4Relight
 **/__pycache__/
 vis_*/
-src/
+src/
+weights/

app.py

@@ -10,6 +10,7 @@ from pytorch_lightning import seed_everything
 import numpy as np
 import os
 import rembg
+from huggingface_hub import snapshot_download
 
 import sys
 from loguru import logger
@@ -92,6 +93,11 @@ description = \
 
 set_loggers("INFO")
 device = "cuda" if torch.cuda.is_available() else "cpu"
+
+# Download from model
+logger.info(f"Downloading Models...")
+snapshot_download(repo_id="LittleFrog/IntrinsicAnything", local_dir="weights")
+
 logger.info(f"Loading Models...")
 model_dict = {
     "Albedo": InferenceModel(ckpt_path="weights/albedo", 

weights/albedo/checkpoints/last.ckpt

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a5fa7a1caa7e1e3818119cd9a2e8715ee7b86a77fa66447cc4b0767d8ab550f8
-size 15458840153

weights/albedo/configs/albedo_project.yaml

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:c8703f33e53258c5597934128f94ae01c4e2c7c0dce84e3c90028ff32fd10719
-size 2526

weights/specular/checkpoints/last.ckpt

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:361b7c3824f4603c4137657ff2fc8a127e0972954d425b440516522f034de7a0
-size 15458847705

weights/specular/configs/specular_project.yaml

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7f1b8dc5694612815974312d64403afcdad9c089b350ab6701b5bb1e3149f8e8
-size 2532