fix:Update app

app.py

@@ -42,8 +42,8 @@ def generate_3d(image, seed=-1,
         seed = np.random.randint(0, MAX_SEED)
     
     image = pipeline.preprocess_image(image, resolution=1024)
-    #normal_image = normal_predictor(image, resolution=768, match_input_resolution=True, data_type='object')
-    normal_image = image
+    normal_image = normal_predictor(image, resolution=768, match_input_resolution=True, data_type='object')
+
     outputs = pipeline.run(
         normal_image,
         seed=seed,
@@ -222,9 +222,9 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
 
 if __name__ == "__main__":
     # Initialize pipeline
-    pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
+    pipeline = TrellisImageTo3DPipeline.from_pretrained("Stable-X/trellis-normal-v0-1")
     pipeline.cuda()
-    
+
     # Initialize normal predictor
     normal_predictor = torch.hub.load("hugoycj/StableNormal", "StableNormal_turbo", trust_repo=True, yoso_version='yoso-normal-v1-8-1')