Spaces:
Runtime error
Runtime error
Upload gradio_app.py
Browse files- gradio_app.py +2 -2
gradio_app.py
CHANGED
@@ -52,7 +52,7 @@ bg_remover = Remover()
|
|
52 |
spar3d_model = SPAR3D.from_pretrained(
|
53 |
"stabilityai/stable-point-aware-3d",
|
54 |
config_name="config.yaml",
|
55 |
-
weight_name="model.safetensors"
|
56 |
token=HF_TOKEN
|
57 |
).eval().to(device)
|
58 |
|
@@ -158,7 +158,7 @@ def forward_model(batch, system, guidance_scale=3.0, seed=0, device="cuda"):
|
|
158 |
|
159 |
@spaces.GPU
|
160 |
@torch.inference_mode()
|
161 |
-
def generate_and_process_3d(image: Image.Image) -> tuple[str | None,
|
162 |
"""Generate image from prompt and convert to 3D model."""
|
163 |
|
164 |
# Generate random seed
|
|
|
52 |
spar3d_model = SPAR3D.from_pretrained(
|
53 |
"stabilityai/stable-point-aware-3d",
|
54 |
config_name="config.yaml",
|
55 |
+
weight_name="model.safetensors",
|
56 |
token=HF_TOKEN
|
57 |
).eval().to(device)
|
58 |
|
|
|
158 |
|
159 |
@spaces.GPU
|
160 |
@torch.inference_mode()
|
161 |
+
def generate_and_process_3d(image: Image.Image) -> tuple[str | None, str | None]:
|
162 |
"""Generate image from prompt and convert to 3D model."""
|
163 |
|
164 |
# Generate random seed
|