Upload app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,8 @@ import spaces
|
|
3 |
from gradio_litmodel3d import LitModel3D
|
4 |
|
5 |
import os
|
6 |
-
|
|
|
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>
|