Update app.py

app.py

@@ -4,6 +4,8 @@ from gradio_litmodel3d import LitModel3D
 import os
 import shutil
 os.environ['SPCONV_ALGO'] = 'native'
+from typing import *
+import imageio
 from easydict import EasyDict as edict
 from PIL import Image, ImageOps
 from trellis.pipelines import TrellisImageTo3DPipeline
@@ -51,10 +53,16 @@ def start_session(req: gr.Request):
 def end_session(req: gr.Request):
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     shutil.rmtree(user_dir)
+
+@spaces.GPU
 def preprocess_image(
     image: Image.Image,
     prompt: str = "",
     negative_prompt: str = "",
+    image=image,
+    num_inference_steps=num_steps,
+    controlnet_conditioning_scale=controlnet_conditioning_scale,
+    guidance_scale=guidance_scale,
     style_name: str = "",
     num_steps: int = 25,
     guidance_scale: float = 5,
@@ -140,6 +148,10 @@ def get_seed(randomize_seed: bool, seed: int) -> int:
 @spaces.GPU
 def image_to_3d(
     image: Image.Image,
+    seed: int,
+    ss_guidance_strength: float,
+    ss_sampling_steps: int,
+    slat_guidance_strength: float,
     slat_sampling_steps: int,
     req: gr.Request,
 ) -> Tuple[dict, str]:
@@ -151,6 +163,10 @@ def image_to_3d(
         preprocess_image=False,
         sparse_structure_sampler_params={
             "steps": ss_sampling_steps,
+            "cfg_strength": ss_guidance_strength,
+        },
+        slat_sampler_params={
+            "steps": slat_sampling_steps,
             "cfg_strength": slat_guidance_strength,
         },
     )