Update app and requirements

app.py

@@ -46,7 +46,7 @@ class Geowizard(object):
     '''
 
     def __init__(self):
-        self.model = DepthNormalEstimationPipeline.from_pretrained("weights/Geowizard/", torch_dtype=torch.float16)
+        self.model = DepthNormalEstimationPipeline.from_pretrained("lemonaddie/Geowizard", torch_dtype=torch.float16)
 
     def cuda(self):
        self.model.cuda()
@@ -103,7 +103,7 @@ class Marigold(Geowizard):
     '''
 
     def __init__(self):
-        self.model= diffusers.MarigoldNormalsPipeline.from_pretrained("weights/marigold-normals-v0-1", torch_dtype=torch.float16)
+        self.model= diffusers.MarigoldNormalsPipeline.from_pretrained("prs-eth/marigold-normals-v0-1", torch_dtype=torch.float16)
 
 
     @torch.no_grad()
@@ -131,9 +131,9 @@ class StableNormal(Geowizard):
     '''
 
     def __init__(self):
-        x_start_pipeline = YOSONormalsPipeline.from_pretrained('/workspace/code/InverseRendering/StableNormal/weights/yoso-normal-v0-2', 
+        x_start_pipeline = YOSONormalsPipeline.from_pretrained('Stable-X/yoso-normal-v0-2',  trust_remote_code=True,
                                                                 variant="fp16", torch_dtype=torch.float16)
-        self.model = StableNormalPipeline.from_pretrained('/workspace/code/InverseRendering/StableNormal/weights/stable-normal-v0-1',
+        self.model = StableNormalPipeline.from_pretrained('Stable-X/stable-normal-v0-1', trust_remote_code=True,
                                                            variant="fp16",  torch_dtype=torch.float16,
                                                           scheduler=HEURI_DDIMScheduler(prediction_type='sample', 
                                                                               beta_start=0.00085, beta_end=0.0120, 

requirements.txt

@@ -109,7 +109,8 @@ sympy==1.12.1
 tokenizers==0.15.2
 tomlkit==0.12.0
 toolz==0.12.1
-torch==2.2.0
+torch==2.0.1
+torchvision==v0.15.2
 tqdm==4.66.4
 transformers==4.36.1
 trimesh==4.0.5

requirements_min.txt

@@ -13,6 +13,7 @@ diffusers>=0.28.0
 matplotlib==3.8.2
 scipy==1.11.4
 torch==2.0.1
+torchvision==v0.15.2
 transformers==4.36.1
 xformers==0.0.21
 opencv-python-headless==4.8.1.78