update

app.py

@@ -37,19 +37,19 @@ os.makedirs(TMP_DIR, exist_ok=True)
 
 RBG_REMOVER = RembgRemover()
 SAM_PREDICTOR = SAMPredictor(model_type="vit_h")
-DELIGHT = DelightingModel()
-IMAGESR_MODEL = ImageRealESRGAN(outscale=4)
-PIPELINE = TrellisImageTo3DPipeline.from_pretrained(
-    "JeffreyXiang/TRELLIS-image-large"
-)
-# PIPELINE.cuda()
+# DELIGHT = DelightingModel()
+# IMAGESR_MODEL = ImageRealESRGAN(outscale=4)
+# PIPELINE = TrellisImageTo3DPipeline.from_pretrained(
+#     "JeffreyXiang/TRELLIS-image-large"
+# )
+# # PIPELINE.cuda()
 
 IMAGE_BUFFER = {}
-SEG_CHECKER = ImageSegChecker(GPT_CLIENT)
-GEO_CHECKER = MeshGeoChecker(GPT_CLIENT)
-AESTHETIC_CHECKER = ImageAestheticChecker()
-CHECKERS = [GEO_CHECKER, SEG_CHECKER, AESTHETIC_CHECKER]
-URDF_CONVERTOR = URDFGenerator(GPT_CLIENT, render_view_num=4)
+# SEG_CHECKER = ImageSegChecker(GPT_CLIENT)
+# GEO_CHECKER = MeshGeoChecker(GPT_CLIENT)
+# AESTHETIC_CHECKER = ImageAestheticChecker()
+# CHECKERS = [GEO_CHECKER, SEG_CHECKER, AESTHETIC_CHECKER]
+# URDF_CONVERTOR = URDFGenerator(GPT_CLIENT, render_view_num=4)
 
 
 def start_session(req: gr.Request) -> None:
@@ -431,7 +431,7 @@ with gr.Blocks(
             slat_guidance_strength,
             slat_sampling_steps,
             gr.State(lambda: IMAGE_BUFFER),
-            gr.State(lambda: PIPELINE),
+            # gr.State(lambda: PIPELINE),
             gr.State(lambda: TMP_DIR),
             image_seg_sam,
             is_samimage,
@@ -448,8 +448,8 @@ with gr.Blocks(
             output_buf,
             project_delight,
             gr.State(lambda: TMP_DIR),
-            gr.State(lambda: DELIGHT),
-            gr.State(lambda: IMAGESR_MODEL),
+            # gr.State(lambda: DELIGHT),
+            # gr.State(lambda: IMAGESR_MODEL),
         ],
         outputs=[
             model_output_mesh,
@@ -472,9 +472,9 @@ with gr.Blocks(
             mass_range_text,
             asset_version_text,
             gr.State(lambda: TMP_DIR),
-            gr.State(lambda: URDF_CONVERTOR),
+            # gr.State(lambda: URDF_CONVERTOR),
             gr.State(lambda: IMAGE_BUFFER),
-            gr.State(lambda: CHECKERS),
+            # gr.State(lambda: CHECKERS),
         ],
         outputs=[
             download_urdf,