From e72ff776c28ad22ebb8ddc2590fb2ca718118f9a Mon Sep 17 00:00:00 2001
From: BIRK Renaud <renaud.birk@imt-atlantique.net>
Date: Wed, 5 Feb 2025 16:06:56 +0100
Subject: [PATCH] fix
---
main.py | 16 ++++++++++------
1 file changed, 10 insertions(+), 6 deletions(-)
diff --git a/main.py b/main.py
index 83d527a..41f526c 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")
--
GitLab