cavargas10 commited on
Commit
9c3e08e
·
verified ·
1 Parent(s): adb9665

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
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__':