Spaces:
Paused
Paused
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
|