Ai_spatial_modeling

Running on Zero

arabago96 commited on May 29

Commit

10392b8

verified ·

1 Parent(s): 2b77a16

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -429,7 +429,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
 # Launch the Gradio app
 if __name__ == "__main__":
-    pipeline = TrellisImageTo3DPipeline.from_pretrained("arabago96/TRELLIS-backup")
     pipeline.cuda()
     try:
         pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))    # Preload rembg

 # Launch the Gradio app
 if __name__ == "__main__":
+    pipeline = TrellisImageTo3DPipeline.from_pretrained("arabago96/ASM-model")
     pipeline.cuda()
     try:
         pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))    # Preload rembg