Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -1,17 +1,22 @@
|
|
1 |
import spaces
|
2 |
import gradio as gr
|
3 |
import os
|
|
|
|
|
4 |
import time
|
5 |
-
import
|
|
|
6 |
import numpy as np
|
7 |
-
|
|
|
8 |
from huggingface_hub import snapshot_download
|
|
|
9 |
from accelerate.utils import set_seed
|
10 |
-
import trimesh
|
11 |
|
12 |
-
from src.utils.data_utils import get_colored_mesh_composition,
|
13 |
-
from src.utils.
|
14 |
from src.pipelines.pipeline_partcrafter import PartCrafterPipeline
|
|
|
15 |
from src.models.briarmbg import BriaRMBG
|
16 |
|
17 |
# Constants
|
|
|
1 |
import spaces
|
2 |
import gradio as gr
|
3 |
import os
|
4 |
+
import sys
|
5 |
+
from glob import glob
|
6 |
import time
|
7 |
+
from typing import Any, Union
|
8 |
+
|
9 |
import numpy as np
|
10 |
+
import torch
|
11 |
+
import trimesh
|
12 |
from huggingface_hub import snapshot_download
|
13 |
+
from PIL import Image
|
14 |
from accelerate.utils import set_seed
|
|
|
15 |
|
16 |
+
from src.utils.data_utils import get_colored_mesh_composition, scene_to_parts, load_surfaces
|
17 |
+
from src.utils.render_utils import render_views_around_mesh, render_normal_views_around_mesh, make_grid_for_images_or_videos, export_renderings
|
18 |
from src.pipelines.pipeline_partcrafter import PartCrafterPipeline
|
19 |
+
from src.utils.image_utils import prepare_image
|
20 |
from src.models.briarmbg import BriaRMBG
|
21 |
|
22 |
# Constants
|