update

app.py

@@ -35,21 +35,21 @@ TMP_DIR = os.path.join(
 )
 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()
+# 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()
 
-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)
+# 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)
 
 
 def start_session(req: gr.Request) -> None:
@@ -233,11 +233,11 @@ with gr.Blocks(
                         )
                     ],
                     inputs=[image_prompt],
-                    fn=partial(
-                        preprocess_image_fn,
-                        model=RBG_REMOVER,
-                        buffer=IMAGE_BUFFER,
-                    ),
+                    # fn=partial(
+                    #     preprocess_image_fn,
+                    #     model=RBG_REMOVER,
+                    #     buffer=IMAGE_BUFFER,
+                    # ),
                     outputs=[image_prompt],
                     run_on_click=True,
                     examples_per_page=32,
@@ -253,11 +253,11 @@ with gr.Blocks(
                         )
                     ],
                     inputs=[image_prompt_sam],
-                    fn=partial(
-                        preprocess_sam_image_fn,
-                        buffer=IMAGE_BUFFER,
-                        model=SAM_PREDICTOR,
-                    ),
+                    # fn=partial(
+                    #     preprocess_sam_image_fn,
+                    #     buffer=IMAGE_BUFFER,
+                    #     model=SAM_PREDICTOR,
+                    # ),
                     outputs=[image_prompt_sam],
                     run_on_click=True,
                     examples_per_page=32,
@@ -305,7 +305,7 @@ with gr.Blocks(
     )
 
     image_prompt.upload(
-        partial(preprocess_image_fn, model=RBG_REMOVER, buffer=IMAGE_BUFFER),
+        # partial(preprocess_image_fn, model=RBG_REMOVER, buffer=IMAGE_BUFFER),
         inputs=[image_prompt],
         outputs=[image_prompt],
     )
@@ -354,9 +354,9 @@ with gr.Blocks(
     )
 
     image_prompt_sam.upload(
-        partial(
-            preprocess_sam_image_fn, buffer=IMAGE_BUFFER, model=SAM_PREDICTOR
-        ),
+        # partial(
+        #     preprocess_sam_image_fn, buffer=IMAGE_BUFFER, model=SAM_PREDICTOR
+        # ),
         inputs=[image_prompt_sam],
         outputs=[image_prompt_sam],
     )
@@ -407,7 +407,7 @@ with gr.Blocks(
             image_prompt_sam,
             selected_points,
             fg_bg_radio,
-            gr.State(lambda: SAM_PREDICTOR),
+            # gr.State(lambda: SAM_PREDICTOR),
         ],
         [image_mask_sam, image_seg_sam],
     )
@@ -430,8 +430,8 @@ with gr.Blocks(
             ss_sampling_steps,
             slat_guidance_strength,
             slat_sampling_steps,
-            gr.State(lambda: IMAGE_BUFFER),
-            gr.State(lambda: PIPELINE),
+            # gr.State(lambda: IMAGE_BUFFER),
+            # 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: IMAGE_BUFFER),
-            gr.State(lambda: CHECKERS),
+            # gr.State(lambda: URDF_CONVERTOR),
+            # gr.State(lambda: IMAGE_BUFFER),
+            # gr.State(lambda: CHECKERS),
         ],
         outputs=[
             download_urdf,