TRELLIS

Running on Zero

gensym commited on 27 days ago

Commit

a8e79b5

1 Parent(s): 639772e

debugging

Files changed (1) hide show

app.py CHANGED Viewed

@@ -275,13 +275,14 @@ cleanup_thread.start()
 @spaces.GPU
 def preload():
-    pass
 # Launch the Gradio app
 if __name__ == "__main__":
     pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
-    pipeline.cuda()
     try:
         pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))    # Preload rembg
     except:

 @spaces.GPU
 def preload():
+    pipeline.cuda()
 # Launch the Gradio app
 if __name__ == "__main__":
+    global pipeline
     pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
+    preload()
     try:
         pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))    # Preload rembg
     except: