debugging

app.py

@@ -273,13 +273,17 @@ def cleanup_tmp_dir():
 cleanup_thread = threading.Thread(target=cleanup_tmp_dir)
 cleanup_thread.start()
 
-
-# Launch the Gradio app
-if __name__ == "__main__":
+@spaces.GPU
+def preload():
+    global pipeline
     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:
         pass
+
+# Launch the Gradio app
+if __name__ == "__main__":
+    preload()
     demo.launch()