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>
|