Update app.py

app.py

@@ -114,67 +114,6 @@ STEP4_HEADER = """
 
 """
 
-# Code for TripoSR
-"""
-# These part of the code (check_input_image and preprocess were taken from https://huggingface.co/spaces/stabilityai/TripoSR/blob/main/app.py)
-if torch.cuda.is_available():
-    device = "cuda:0"
-else:
-    device = "cpu"
-
-model = TSR.from_pretrained(
-    "stabilityai/TripoSR",
-    config_name="config.yaml",
-    weight_name="model.ckpt",
-)
-model.renderer.set_chunk_size(131072)
-model.to(device)
-
-rembg_session = rembg.new_session()
-
-
-def check_input_image(input_image):
-    if input_image is None:
-        raise gr.Error("No image uploaded!")
-
-
-def preprocess(input_image, do_remove_background, foreground_ratio):
-    def fill_background(image):
-        image = np.array(image).astype(np.float32) / 255.0
-        image = image[:, :, :3] * image[:, :, 3:4] + (1 - image[:, :, 3:4]) * 0.5
-        image = Image.fromarray((image * 255.0).astype(np.uint8))
-        return image
-
-    if do_remove_background:
-        image = input_image.convert("RGB")
-        image = remove_background(image, rembg_session)
-        image = resize_foreground(image, foreground_ratio)
-        image = fill_background(image)
-    else:
-        image = input_image
-        if image.mode == "RGBA":
-            image = fill_background(image)
-    return image
-
-
-@spaces.GPU
-def generate(image, mc_resolution, formats=["obj", "glb"]):
-    scene_codes = model(image, device=device)
-    mesh = model.extract_mesh(scene_codes, resolution=mc_resolution)[0]
-    mesh = to_gradio_3d_orientation(mesh)
-
-    mesh_path_glb = tempfile.NamedTemporaryFile(suffix=f".glb", delete=False)
-    mesh.export(mesh_path_glb.name)
-
-    mesh_path_obj = tempfile.NamedTemporaryFile(suffix=f".obj", delete=False)
-    mesh.apply_scale([-1, 1, 1])  # Otherwise the visualized .obj will be flipped
-    mesh.export(mesh_path_obj.name)
-    
-    return mesh_path_obj.name, mesh_path_glb.name
-"""
-
-
-
 ###############################################################################
 # Configuration for InstantMesh
 # All this code is from https://huggingface.co/spaces/TencentARC/InstantMesh/blob/main/app.py
@@ -186,7 +125,7 @@ if cuda_path:
 else:
     print("CUDA installation not found")
 
-config_path = 'configs/instant-mesh-large.yaml'
+config_path = 'instantmesh/configs/instant-mesh-large.yaml'
 config = OmegaConf.load(config_path)
 config_name = os.path.basename(config_path).replace('.yaml', '')
 model_config = config.model_config