zhuhai111 commited on
Commit
0435c49
·
verified ·
1 Parent(s): ddbac99

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -3,7 +3,8 @@ import spaces
3
  from gradio_litmodel3d import LitModel3D
4
 
5
  import os
6
- import shutil
 
7
  os.environ['SPCONV_ALGO'] = 'native'
8
  from typing import *
9
  import torch
@@ -116,8 +117,7 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
116
  )
117
 
118
  with gr.Row():
119
- gr.Markdown("""
120
- <p align="center">
121
  <a title="Website" href="https://stable-x.github.io/Hi3DGen/" target="_blank" rel="noopener noreferrer" style="display: inline-block;">
122
  <img src="https://www.obukhov.ai/img/badges/badge-website.svg">
123
  </a>
 
3
  from gradio_litmodel3d import LitModel3D
4
 
5
  import os
6
+ # Set CPU_ONLY environment variable to force CPU mode
7
+ os.environ['CPU_ONLY'] = '1'
8
  os.environ['SPCONV_ALGO'] = 'native'
9
  from typing import *
10
  import torch
 
117
  )
118
 
119
  with gr.Row():
120
+ gr.Markdown("""<p align="center">
 
121
  <a title="Website" href="https://stable-x.github.io/Hi3DGen/" target="_blank" rel="noopener noreferrer" style="display: inline-block;">
122
  <img src="https://www.obukhov.ai/img/badges/badge-website.svg">
123
  </a>