Update app.py

app.py

@@ -164,22 +164,13 @@ class StableNormal(Geowizard):
 
 class YosoNormal(Geowizard):
     def __init__(self):
-        self.model = YOSONormalsPipeline.from_pretrained('Stable-X/yoso-normal-v0-3',  trust_remote_code=True,
-                                                                variant="fp16", torch_dtype=torch.float16, t_start=0)
-        
-        # two stage concat
-        self.model.x_start_pipeline = x_start_pipeline
-        self.model.x_start_pipeline.to('cuda', torch.float16)
-        self.model.prior.to('cuda', torch.float16)
-
+        self.model = torch.hub.load("hugoycj/StableNormal", "StableNormal_turbo", trust_repo=True, yoso_version='yoso-normal-v1-8-1')
+        self.model.to('cuda', torch.float16)
 
     @torch.no_grad()
     def __call__(self, img, image_resolution=768):
-        pipe_out = self.model(Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB)))
-        pred_normal = pipe_out.prediction[0]
-        pred_normal = (pred_normal + 1) / 2 * 255
-        pred_normal = pred_normal.astype(np.uint8)
-
+        pred_normal = self.model(Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB)))
+        pred_normal = np.asarray(pred_normal)
         return pred_normal
 
     def to(self, device):