Update app.py

app.py

@@ -272,13 +272,16 @@ def create_gradio_interface():
 
         extract_glb_btn.click(
             extract_glb,
-            inputs=[state, mesh_simplify, texture_size],
+            inputs=[ mesh_simplify, texture_size],
             outputs=[glb_output]
         )
 
-    return demo
-
-# Launch the interface
+# Launch the Gradio app
 if __name__ == "__main__":
-    interface = create_gradio_interface()
-    interface.launch()
+    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
+    demo.launch()