Upload app.py with huggingface_hub

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-# import spaces  # Not needed on dedicated GPU
+# import spaces  # Not using ZeroGPU on dedicated L40S
 from gradio_litmodel3d import LitModel3D
 
 import os
@@ -253,15 +253,14 @@ def split_image(image: Image.Image) -> List[Image.Image]:
 
 
 
-# Initialize pipeline (dedicated GPU - load at startup)
+# ===== Load pipeline at module level (dedicated L40S GPU) =====
 pipeline = TrellisImageTo3DPipeline.from_pretrained("microsoft/TRELLIS-image-large")
 pipeline.cuda()
 try:
     pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))    # Preload rembg
 except:
     pass
-print("Pipeline loaded successfully")
-
+print("Pipeline loaded successfully on dedicated GPU")
 
 with gr.Blocks(delete_cache=(600, 600)) as demo:
     gr.Markdown("""
@@ -414,5 +413,4 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
 
 # Launch the Gradio app
 if __name__ == "__main__":
-        pass
     demo.launch()