Update app.py

app.py

@@ -164,17 +164,12 @@ def activate_button() -> gr.Button:
 def deactivate_button() -> gr.Button:
     return gr.Button(interactive=False)
 
-def load_model():
+with gr.Blocks() as demo:
     pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")                                            
     try:
         pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))    # Preload rembg
     except:
         pass
-    model = {"pipeline":pipeline}
-    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.