Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+methodology.png filter=lfs diff=lfs merge=lfs -text

vaas/inference/pipeline.py

@@ -1,9 +1,10 @@
 import os
 from typing import Dict, Union
-
-import torch
 from PIL import Image
-import torchvision.transforms as T
+
+from vaas.utils.helpers import require_torch_or_exit
+torch, T = require_torch_or_exit()
+
 
 from vaas.fx.fx_model import FxViT
 from vaas.px.px_model import PatchConsistencySegformer
@@ -19,6 +20,7 @@ hf_logging.set_verbosity_error()
 from huggingface_hub import hf_hub_download
 
 
+
 class VAASPipeline:
     def __init__(
         self,