Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -117,7 +117,7 @@ def get_random_hex():
|
|
117 |
random_hex = random_bytes.hex()
|
118 |
return random_hex
|
119 |
|
120 |
-
@spaces.GPU(duration=
|
121 |
def run_full(image: str, seed: int = 0, num_inference_steps: int = 50, guidance_scale: float = 7.5, simplify: bool = True, target_face_num: int = DEFAULT_FACE_NUMBER, req=None):
|
122 |
try:
|
123 |
image_seg = prepare_image(image, bg_color=np.array([1.0, 1.0, 1.0]), rmbg_net=rmbg_net)
|
@@ -303,7 +303,7 @@ def run_segmentation(image: str):
|
|
303 |
logger.error(f"Error in run_segmentation: {str(e)}")
|
304 |
raise
|
305 |
|
306 |
-
@spaces.GPU(duration=
|
307 |
@torch.no_grad()
|
308 |
def image_to_3d(
|
309 |
image, # Changed to accept FileData dict or PIL Image
|
@@ -357,7 +357,7 @@ def image_to_3d(
|
|
357 |
logger.error(f"Error in image_to_3d: {str(e)}")
|
358 |
raise
|
359 |
|
360 |
-
@spaces.GPU(duration=
|
361 |
@torch.no_grad()
|
362 |
def run_texture(image: Image, mesh_path: str, seed: int, req: gr.Request):
|
363 |
try:
|
|
|
117 |
random_hex = random_bytes.hex()
|
118 |
return random_hex
|
119 |
|
120 |
+
@spaces.GPU(duration=30)
|
121 |
def run_full(image: str, seed: int = 0, num_inference_steps: int = 50, guidance_scale: float = 7.5, simplify: bool = True, target_face_num: int = DEFAULT_FACE_NUMBER, req=None):
|
122 |
try:
|
123 |
image_seg = prepare_image(image, bg_color=np.array([1.0, 1.0, 1.0]), rmbg_net=rmbg_net)
|
|
|
303 |
logger.error(f"Error in run_segmentation: {str(e)}")
|
304 |
raise
|
305 |
|
306 |
+
@spaces.GPU(duration=30)
|
307 |
@torch.no_grad()
|
308 |
def image_to_3d(
|
309 |
image, # Changed to accept FileData dict or PIL Image
|
|
|
357 |
logger.error(f"Error in image_to_3d: {str(e)}")
|
358 |
raise
|
359 |
|
360 |
+
@spaces.GPU(duration=30)
|
361 |
@torch.no_grad()
|
362 |
def run_texture(image: Image, mesh_path: str, seed: int, req: gr.Request):
|
363 |
try:
|