Steven18 commited on
Commit
f0c50fc
·
1 Parent(s): 6b314ed

test for api generate

Browse files
Files changed (1) hide show
  1. app.py +52 -28
app.py CHANGED
@@ -280,20 +280,38 @@ def quick_generate_glb(
280
  texture_size: int,
281
  req: gr.Request,
282
  ) -> Tuple[str, str]:
283
- seed = get_seed(True, 0)
284
  state, _ = image_to_3d(
285
  image=image,
286
- multiimages=[],
287
- is_multiimage=False,
288
  seed=seed,
289
- ss_guidance_strength=7.5,
290
- ss_sampling_steps=12,
291
- slat_guidance_strength=3.0,
292
- slat_sampling_steps=12,
293
- multiimage_algo="stochastic",
294
- req=None
295
  )
296
- return extract_glb(state, mesh_simplify=0.95, texture_size=1024, req=None)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
297
 
298
 
299
 
@@ -343,6 +361,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
343
  extract_gs_btn = gr.Button("Extract Gaussian", interactive=False)
344
  generate_glb_btn = gr.Button("Quick Generate GLB in layout")
345
  quick_generate_glb_btn = gr.Button("Quick Generate GLB with download")
 
346
  gr.Markdown("""
347
  *NOTE: Gaussian file can be very large (~50MB), it will take a while to display and download.*
348
  """)
@@ -474,24 +493,29 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
474
  outputs=[download_glb]
475
  )
476
 
477
- # quick_generate_glb_btn.click(
478
- # fn=quick_generate_glb,
479
- # inputs=[
480
- # image_input,
481
- # gr.State([]), # multiimages
482
- # gr.State(False), # is_multiimage
483
- # gr.State(0), # seed
484
- # gr.State(7.5), # ss_guidance_strength
485
- # gr.State(12), # ss_sampling_steps
486
- # gr.State(3.0), # slat_guidance_strength
487
- # gr.State(12), # slat_sampling_steps
488
- # gr.State("stochastic"), # multiimage_algo
489
- # gr.State(0.95), # mesh_simplify
490
- # gr.State(1024), # texture_size
491
- # ],
492
- # outputs=[model_output, glb_download],
493
- # api_name="quick_generate_glb"
494
- # )
 
 
 
 
 
495
 
496
 
497
 
 
280
  texture_size: int,
281
  req: gr.Request,
282
  ) -> Tuple[str, str]:
 
283
  state, _ = image_to_3d(
284
  image=image,
285
+ multiimages=multiimages,
286
+ is_multiimage=is_multiimage,
287
  seed=seed,
288
+ ss_guidance_strength=ss_guidance_strength,
289
+ ss_sampling_steps=ss_sampling_steps,
290
+ slat_guidance_strength=slat_guidance_strength,
291
+ slat_sampling_steps=slat_sampling_steps,
292
+ multiimage_algo=multiimage_algo,
293
+ req=req
294
  )
295
+ return extract_glb(state, mesh_simplify=mesh_simplify, texture_size=texture_size, req=req)
296
+
297
+ @spaces.GPU(api_name="quick_glb_dl")
298
+ def quick_glb_dl(image: Image.Image, req: gr.Request) -> str:
299
+ seed = get_seed(True, 0)
300
+ state, _ = image_to_3d(image, [], False, seed, 7.5, 12, 3.0, 12, "stochastic", req)
301
+ glb_path, _ = extract_glb(state, 0.95, 1024, req)
302
+ return glb_path
303
+
304
+ def test_for_api_gen(image: Image.Image) -> Image.Image:
305
+ """
306
+ bilibili .
307
+
308
+ Args:
309
+ image (Image.Image): The input imagein hererererer.
310
+
311
+ Returns:
312
+ Image.Image: The preprocessed image no processs.
313
+ """
314
+ return image
315
 
316
 
317
 
 
361
  extract_gs_btn = gr.Button("Extract Gaussian", interactive=False)
362
  generate_glb_btn = gr.Button("Quick Generate GLB in layout")
363
  quick_generate_glb_btn = gr.Button("Quick Generate GLB with download")
364
+ quick_generate_glb_easy_btn = gr.Button("Easy Quick Generate GLB with download")
365
  gr.Markdown("""
366
  *NOTE: Gaussian file can be very large (~50MB), it will take a while to display and download.*
367
  """)
 
493
  outputs=[download_glb]
494
  )
495
 
496
+ quick_generate_glb_btn.click(
497
+ fn=quick_generate_glb,
498
+ inputs=[
499
+ image_prompt,
500
+ multiimage_prompt,
501
+ is_multiimage,
502
+ seed,
503
+ ss_guidance_strength,
504
+ ss_sampling_steps,
505
+ slat_guidance_strength,
506
+ slat_sampling_steps,
507
+ multiimage_algo,
508
+ mesh_simplify,
509
+ texture_size,
510
+ ],
511
+ outputs=[model_output, download_glb],
512
+ )
513
+
514
+ quick_generate_glb_easy_btn.click(
515
+ fn=quick_glb_dl,
516
+ inputs=[image_prompt],
517
+ outputs=[download_glb],
518
+ )
519
 
520
 
521