Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
import shlex
|
2 |
import subprocess
|
3 |
|
|
|
4 |
subprocess.run(shlex.split("pip install pip==24.0"), check=True)
|
5 |
subprocess.run(
|
6 |
shlex.split(
|
@@ -13,9 +14,12 @@ subprocess.run(
|
|
13 |
), check=True
|
14 |
)
|
15 |
|
|
|
16 |
if __name__ == "__main__":
|
17 |
from huggingface_hub import snapshot_download
|
|
|
18 |
snapshot_download("public-data/Unique3D", repo_type="model", local_dir="./ckpt")
|
|
|
19 |
import os
|
20 |
import sys
|
21 |
sys.path.append(os.curdir)
|
@@ -23,11 +27,13 @@ if __name__ == "__main__":
|
|
23 |
torch.set_float32_matmul_precision('medium')
|
24 |
torch.backends.cuda.matmul.allow_tf32 = True
|
25 |
torch.set_grad_enabled(False)
|
|
|
26 |
import fire
|
27 |
import gradio as gr
|
28 |
from gradio_app.gradio_3dgen import create_ui as create_3d_ui
|
29 |
from gradio_app.all_models import model_zoo
|
30 |
|
|
|
31 |
_TITLE = '''Unique3D: High-Quality and Efficient 3D Mesh Generation from a Single Image'''
|
32 |
_DESCRIPTION = '''
|
33 |
<div>
|
@@ -52,9 +58,8 @@ def launch():
|
|
52 |
with gr.Column(scale=1):
|
53 |
gr.Markdown('# ' + _TITLE)
|
54 |
gr.Markdown(_DESCRIPTION)
|
55 |
-
|
56 |
create_3d_ui("wkl")
|
57 |
-
|
58 |
demo.queue().launch(share=True)
|
59 |
|
60 |
if __name__ == '__main__':
|
|
|
1 |
import shlex
|
2 |
import subprocess
|
3 |
|
4 |
+
|
5 |
subprocess.run(shlex.split("pip install pip==24.0"), check=True)
|
6 |
subprocess.run(
|
7 |
shlex.split(
|
|
|
14 |
), check=True
|
15 |
)
|
16 |
|
17 |
+
|
18 |
if __name__ == "__main__":
|
19 |
from huggingface_hub import snapshot_download
|
20 |
+
|
21 |
snapshot_download("public-data/Unique3D", repo_type="model", local_dir="./ckpt")
|
22 |
+
|
23 |
import os
|
24 |
import sys
|
25 |
sys.path.append(os.curdir)
|
|
|
27 |
torch.set_float32_matmul_precision('medium')
|
28 |
torch.backends.cuda.matmul.allow_tf32 = True
|
29 |
torch.set_grad_enabled(False)
|
30 |
+
|
31 |
import fire
|
32 |
import gradio as gr
|
33 |
from gradio_app.gradio_3dgen import create_ui as create_3d_ui
|
34 |
from gradio_app.all_models import model_zoo
|
35 |
|
36 |
+
|
37 |
_TITLE = '''Unique3D: High-Quality and Efficient 3D Mesh Generation from a Single Image'''
|
38 |
_DESCRIPTION = '''
|
39 |
<div>
|
|
|
58 |
with gr.Column(scale=1):
|
59 |
gr.Markdown('# ' + _TITLE)
|
60 |
gr.Markdown(_DESCRIPTION)
|
|
|
61 |
create_3d_ui("wkl")
|
62 |
+
|
63 |
demo.queue().launch(share=True)
|
64 |
|
65 |
if __name__ == '__main__':
|