diff --git a/main.py b/main.py index 83d527abdd633d627ff9cd98e9f0b3f713781180..41f526c049eaf635388c4fda034d0d9b5de8c0da 100644 --- a/main.py +++ b/main.py @@ -2,6 +2,11 @@ from gradio_client import Client, handle_file client = Client("JeffreyXiang/TRELLIS") +result = client.predict( + api_name="/start_session" +) +print(result) + preprocessed_image = client.predict( image=handle_file('https://i.pinimg.com/564x/79/b3/6d/79b36deeb891cf65e41c743f5350d6cf.jpg'), api_name="/preprocess_image" @@ -17,14 +22,13 @@ result = client.predict( image=handle_file(preprocessed_image), multiimages=[], seed=rand_seed, - ss_guidance_strength=rand_seed, - ss_sampling_steps=7.5, - slat_guidance_strength=12, - slat_sampling_steps=3, - multiimage_algo="12", + ss_guidance_strength=7.5, + ss_sampling_steps=12, + slat_guidance_strength=3, + slat_sampling_steps=12, + multiimage_algo="stochastic", api_name="/image_to_3d" ) - print(result) print("image_to_3d ok")