Update app.py

app.py

@@ -103,7 +103,8 @@ class Marigold(Geowizard):
     '''
 
     def __init__(self):
-        self.model= diffusers.MarigoldNormalsPipeline.from_pretrained("prs-eth/marigold-normals-v0-1", torch_dtype=torch.float16)
+        self.model= diffusers.MarigoldNormalsPipeline.from_pretrained("prs-eth/marigold-normals-lcm-v0-1", 
+                                                                      variant="fp16", torch_dtype=torch.float16)
 
 
     @torch.no_grad()
@@ -315,7 +316,6 @@ def run_demo_server(pipe):
                     label="Marigold",
                     type="filepath",
                 )
-        with gr.Row():  
             with gr.Column():
                 Ours_slider = gr.Image(
                     label="StableNormal",
@@ -389,7 +389,6 @@ def main():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     dsine_pipe = DSINE()
     marigold_pipe = Marigold()
-    geowizard_pipe = Geowizard()
     our_pipe = StableNormal()