Upload gradio_app.py

gradio_app.py

@@ -43,35 +43,6 @@ if True:
     subprocess.run(shlex.split("pip install custom_rasterizer-0.1-cp310-cp310-linux_x86_64.whl"), check=True)
 
 
-def get_example_img_list():
-    print('Loading example img list ...')
-    return sorted(glob('./assets/example_images/**/*.png', recursive=True))
-
-
-def get_example_txt_list():
-    print('Loading example txt list ...')
-    txt_list = list()
-    for line in open('./assets/example_prompts.txt', encoding='utf-8'):
-        txt_list.append(line.strip())
-    return txt_list
-
-
-def get_example_mv_list():
-    print('Loading example mv list ...')
-    mv_list = list()
-    root = './assets/example_mv_images'
-    for mv_dir in os.listdir(root):
-        view_list = []
-        for view in ['front', 'back', 'left', 'right']:
-            path = os.path.join(root, mv_dir, f'{view}.png')
-            if os.path.exists(path):
-                view_list.append(path)
-            else:
-                view_list.append(None)
-        mv_list.append(view_list)
-    return mv_list
-
-
 def gen_save_folder(max_size=200):
     os.makedirs(SAVE_DIR, exist_ok=True)
 
@@ -144,8 +115,6 @@ def build_model_viewer_html(save_folder, height=660, width=790, textured=False):
 
 @spaces.GPU(duration=40)
 def _gen_shape(
-    caption=None,
-    image=None,
     mv_image_front=None,
     mv_image_back=None,
     mv_image_left=None,
@@ -158,8 +127,6 @@ def _gen_shape(
     num_chunks=200000,
     randomize_seed: bool = False,
 ):
-    if not MV_MODE and image is None and caption is None:
-        raise gr.Error("Please provide either a caption or an image.")
     if MV_MODE:
         if mv_image_front is None and mv_image_back is None and mv_image_left is None and mv_image_right is None:
             raise gr.Error("Please provide at least one view image.")
@@ -176,7 +143,6 @@ def _gen_shape(
     seed = int(randomize_seed_fn(seed, randomize_seed))
 
     octree_resolution = int(octree_resolution)
-    if caption: print('prompt is', caption)
     save_folder = gen_save_folder()
     stats = {
         'model': {
@@ -184,7 +150,6 @@ def _gen_shape(
             'texgen': f'{args.texgen_model_path}',
         },
         'params': {
-            'caption': caption,
             'steps': steps,
             'guidance_scale': guidance_scale,
             'seed': seed,
@@ -251,8 +216,6 @@ def _gen_shape(
 
 @spaces.GPU(duration=90)
 def generation_all(
-    caption=None,
-    image=None,
     mv_image_front=None,
     mv_image_back=None,
     mv_image_left=None,
@@ -267,8 +230,6 @@ def generation_all(
 ):
     start_time_0 = time.time()
     mesh, image, save_folder, stats, seed = _gen_shape(
-        caption,
-        image,
         mv_image_front=mv_image_front,
         mv_image_back=mv_image_back,
         mv_image_left=mv_image_left,
@@ -316,8 +277,6 @@ def generation_all(
 
 @spaces.GPU(duration=40)
 def shape_generation(
-    caption=None,
-    image=None,
     mv_image_front=None,
     mv_image_back=None,
     mv_image_left=None,
@@ -332,8 +291,6 @@ def shape_generation(
 ):
     start_time_0 = time.time()
     mesh, image, save_folder, stats, seed = _gen_shape(
-        caption,
-        image,
         mv_image_front=mv_image_front,
         mv_image_back=mv_image_back,
         mv_image_left=mv_image_left,
@@ -469,8 +426,6 @@ def build_app():
         btn.click(
             shape_generation,
             inputs=[
-                caption,
-                image,
                 mv_image_front,
                 mv_image_back,
                 mv_image_left,
@@ -496,8 +451,6 @@ def build_app():
         btn_all.click(
             generation_all,
             inputs=[
-                caption,
-                image,
                 mv_image_front,
                 mv_image_back,
                 mv_image_left,
@@ -633,9 +586,6 @@ if __name__ == '__main__':
     border-color: #e5e7eb; order-style: solid; border-width: 1px;'>
     </div>
     """
-    example_is = get_example_img_list()
-    example_ts = get_example_txt_list()
-    example_mvs = get_example_mv_list()
 
     SUPPORTED_FORMATS = ['glb', 'obj', 'ply', 'stl']