Spaces:
Sleeping
Sleeping
Fix: gpu errro
Browse files- gradio_app(original).py +14 -5
gradio_app(original).py
CHANGED
|
@@ -786,6 +786,20 @@ Fast for very complex cases, Standard seldom use.',
|
|
| 786 |
return demo
|
| 787 |
|
| 788 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 789 |
if __name__ == '__main__':
|
| 790 |
import argparse
|
| 791 |
|
|
@@ -919,9 +933,4 @@ if __name__ == '__main__':
|
|
| 919 |
demo = build_app()
|
| 920 |
app = gr.mount_gradio_app(app, demo, path="/")
|
| 921 |
|
| 922 |
-
if ENV == 'Huggingface':
|
| 923 |
-
# for Zerogpu
|
| 924 |
-
from spaces import zero
|
| 925 |
-
zero.startup()
|
| 926 |
-
|
| 927 |
uvicorn.run(app, host=args.host, port=args.port)
|
|
|
|
| 786 |
return demo
|
| 787 |
|
| 788 |
|
| 789 |
+
def prepare_huggingface_env():
|
| 790 |
+
import subprocess, shlex, os
|
| 791 |
+
# wheel install / compile steps
|
| 792 |
+
subprocess.run(shlex.split(
|
| 793 |
+
"pip install custom_rasterizer-0.1-cp310-cp310-linux_x86_64.whl"), check=True)
|
| 794 |
+
os.system("cd ./hy3dpaint/DifferentiableRenderer && bash compile_mesh_painter.sh")
|
| 795 |
+
|
| 796 |
+
# --- function & class definitions (with @spaces.GPU) here ---
|
| 797 |
+
|
| 798 |
+
if ENV == "Huggingface":
|
| 799 |
+
prepare_huggingface_env()
|
| 800 |
+
from spaces import zero
|
| 801 |
+
zero.startup()
|
| 802 |
+
|
| 803 |
if __name__ == '__main__':
|
| 804 |
import argparse
|
| 805 |
|
|
|
|
| 933 |
demo = build_app()
|
| 934 |
app = gr.mount_gradio_app(app, demo, path="/")
|
| 935 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 936 |
uvicorn.run(app, host=args.host, port=args.port)
|