remove pip import

README.md

@@ -4,7 +4,7 @@ emoji: 📸🚀🌟
 colorFrom: red
 colorTo: yellow
 sdk: gradio
-sdk_version: 3.41.1
+sdk_version: 3.44.0
 app_file: app.py
 pinned: true
 license: mit

app.py

@@ -1,7 +1,8 @@
 import os, sys
+import subprocess
 from huggingface_hub import snapshot_download
 
-is_local_run = False
+is_local_run = True
 
 code_dir = snapshot_download("One-2-3-45/code") if not is_local_run else "../code" # , token=os.environ['TOKEN']
 
@@ -11,16 +12,18 @@ elev_est_dir = os.path.abspath(os.path.join(code_dir, "one2345_elev_est"))
 sys.path.append(elev_est_dir)
 
 if not is_local_run:
-    import pip
-    pip.main(['install', elev_est_dir])
+    # import pip
+    # pip.main(['install', elev_est_dir])
     # export TORCH_CUDA_ARCH_LIST="7.0;7.2;8.0;8.6"
     # export IABN_FORCE_CUDA=1
     os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
     os.environ["IABN_FORCE_CUDA"] = "1"
     os.environ["FORCE_CUDA"] = "1"
-    pip.main(["install", "inplace_abn"]) 
+    # pip.main(["install", "inplace_abn"]) 
+    subprocess.run(['pip', 'install', 'inplace_abn'])
     # FORCE_CUDA=1 pip install --no-cache-dir git+https://github.com/mit-han-lab/[email protected]
-    pip.main(["install", "--no-cache-dir", "git+https://github.com/mit-han-lab/[email protected]"])
+    # pip.main(["install", "--no-cache-dir", "git+https://github.com/mit-han-lab/[email protected]"])
+    subprocess.run(['pip', 'install', '--no-cache-dir', 'git+https://github.com/mit-han-lab/[email protected]'])
 
 import shutil
 import torch
@@ -487,7 +490,7 @@ def run_demo(
         gr.Markdown(_DESCRIPTION)
 
         with gr.Row(variant='panel'):
-            with gr.Column(scale=1.2):
+            with gr.Column(scale=6):
                 image_block = gr.Image(type='pil', image_mode='RGBA', height=290, label='Input image', tool=None)
 
                 gr.Examples(
@@ -511,7 +514,7 @@ def run_demo(
                 run_btn = gr.Button('Run Generation', variant='primary', interactive=False)
                 guide_text = gr.Markdown(_USER_GUIDE, visible=True)
                 
-            with gr.Column(scale=.8):
+            with gr.Column(scale=4):
                 with gr.Row():
                     bbox_block = gr.Image(type='pil', label="Bounding box", height=290, interactive=False)
                     sam_block = gr.Image(type='pil', label="SAM output", interactive=False)
@@ -527,11 +530,11 @@ def run_demo(
                 mesh_output = gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0], label="One-2-3-45's Textured Mesh", elem_id="model-3d-out")
         
         with gr.Row(variant='panel'):
-            with gr.Column(scale=0.85):
+            with gr.Column(scale=85):
                 elev_output = gr.Label(label='Estimated elevation (degree, w.r.t. the horizontal plane)')
                 vis_output = gr.Plot(label='Camera poses of the input view (red) and predicted views (blue)', elem_id="plot-out")
                 
-            with gr.Column(scale=1.15):
+            with gr.Column(scale=115):
                 gr.Markdown('Predicted multi-view images')
                 with gr.Row():
                     view_1 = gr.Image(interactive=False, height=200, show_label=False)

requirements.txt

@@ -1,5 +1,5 @@
 --extra-index-url https://download.pytorch.org/whl/cu118
-torch>=2.0.0
+torch==2.0.1
 torchvision
 albumentations>=0.4.3
 opencv-python>=4.5.5.64