Update app.py

app.py

@@ -484,12 +484,12 @@ def main():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
     x_start_pipeline = YOSONormalsPipeline.from_pretrained(
-        'weights/yoso-normal-v0-1', trust_remote_code=True,
+        'Stable-X/yoso-normal-v0-1', trust_remote_code=True,
         t_start=300).to(device)
     dinov2_prior = DINOv2_Encoder(size=672)
     dinov2_prior.to(device)
     
-    pipe = StableNormalPipeline.from_pretrained('weights/stable-normal-v0-1', t_start=300, trust_remote_code=True,
+    pipe = StableNormalPipeline.from_pretrained('Stable-X/stable-normal-v0-1', t_start=300, trust_remote_code=True,
                                                 scheduler=HEURI_DDIMScheduler(prediction_type='sample', 
                                                                               beta_start=0.00085, beta_end=0.0120, 
                                                                               beta_schedule = "scaled_linear"))