alexnasa commited on
Commit
c5402af
·
verified ·
1 Parent(s): 6813a3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -5
app.py CHANGED
@@ -1,17 +1,22 @@
1
  import spaces
2
  import gradio as gr
3
  import os
 
 
4
  import time
5
- import torch
 
6
  import numpy as np
7
- from PIL import Image
 
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, export_renderings
13
- from src.utils.image_utils import prepare_image
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