Spaces:
Running
on
Zero
Running
on
Zero
try spaces.gpu on only run()
Browse files
app.py
CHANGED
@@ -4,13 +4,11 @@ from __future__ import annotations
|
|
4 |
|
5 |
import os
|
6 |
|
7 |
-
import spaces
|
8 |
import gradio as gr
|
9 |
|
10 |
from model import Model
|
11 |
|
12 |
|
13 |
-
@spaces.GPU
|
14 |
def main():
|
15 |
DESCRIPTION = '''# [TEXTure](https://github.com/TEXTurePaper/TEXTurePaper)
|
16 |
|
|
|
4 |
|
5 |
import os
|
6 |
|
|
|
7 |
import gradio as gr
|
8 |
|
9 |
from model import Model
|
10 |
|
11 |
|
|
|
12 |
def main():
|
13 |
DESCRIPTION = '''# [TEXTure](https://github.com/TEXTurePaper/TEXTurePaper)
|
14 |
|
model.py
CHANGED
@@ -9,6 +9,7 @@ from typing import Generator, Optional
|
|
9 |
|
10 |
import gradio as gr
|
11 |
import trimesh
|
|
|
12 |
|
13 |
sys.path.append('TEXTurePaper')
|
14 |
|
@@ -50,6 +51,7 @@ class Model:
|
|
50 |
subprocess.run(shlex.split(f'zip -r {out_path} {mesh_dir}'))
|
51 |
return out_path
|
52 |
|
|
|
53 |
def run(
|
54 |
self, shape_path: str, text: str, seed: int, guidance_scale: float
|
55 |
) -> Generator[tuple[list[str], Optional[str], Optional[str], str], None,
|
|
|
9 |
|
10 |
import gradio as gr
|
11 |
import trimesh
|
12 |
+
import spaces
|
13 |
|
14 |
sys.path.append('TEXTurePaper')
|
15 |
|
|
|
51 |
subprocess.run(shlex.split(f'zip -r {out_path} {mesh_dir}'))
|
52 |
return out_path
|
53 |
|
54 |
+
@spaces.GPU
|
55 |
def run(
|
56 |
self, shape_path: str, text: str, seed: int, guidance_scale: float
|
57 |
) -> Generator[tuple[list[str], Optional[str], Optional[str], str], None,
|