John6666 commited on
Commit
581c8b5
·
verified ·
1 Parent(s): 8464784

Upload gradio_app.py

Browse files
Files changed (1) hide show
  1. 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, Image.Image | 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