Update app.py

app.py

@@ -174,6 +174,7 @@ def load_model():
     return model
 
 with gr.Blocks() as demo:
+    demo.load(fn=load_model, inputs=None, outputs=None)
     gr.Markdown("""
     ## Image to 3D Asset with [TRELLIS](https://trellis3d.github.io/)
     * Upload an image and click "Generate" to create a 3D asset. If the image has alpha channel, it be used as the mask. Otherwise, we use `rembg` to remove the background.
@@ -268,5 +269,4 @@ if __name__ == "__main__":
     #print(f"CUDA Available: {torch.cuda.is_available()}")
     #print(f"CUDA Version: {torch.version.cuda}")
     #print(f"Number of GPUs: {torch.cuda.device_count()}")
-    demo.load(fn=load_model, inputs=None, outputs=None)
     demo.launch(debug=True)