From fbb694c2c5f376a8977fd3427317c51fe9ee48c8 Mon Sep 17 00:00:00 2001
From: BIRK Renaud <renaud.birk@imt-atlantique.net>
Date: Wed, 5 Feb 2025 15:55:50 +0100
Subject: [PATCH] feat: base pipeline
---
main.py | 44 ++++++++++++++++++++++++++++++++++++++++++++
requirements.txt | 1 +
2 files changed, 45 insertions(+)
create mode 100644 main.py
create mode 100644 requirements.txt
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..83d527a
--- /dev/null
+++ b/main.py
@@ -0,0 +1,44 @@
+from gradio_client import Client, handle_file
+
+client = Client("JeffreyXiang/TRELLIS")
+
+preprocessed_image = client.predict(
+ image=handle_file('https://i.pinimg.com/564x/79/b3/6d/79b36deeb891cf65e41c743f5350d6cf.jpg'),
+ api_name="/preprocess_image"
+)
+
+rand_seed = client.predict(
+ randomize_seed=True,
+ seed=0,
+ api_name="/get_seed"
+)
+
+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",
+ api_name="/image_to_3d"
+)
+
+print(result)
+
+print("image_to_3d ok")
+
+client.predict(
+ api_name="/lambda_2"
+)
+
+client.predict(
+ mesh_simplify=None,
+ texture_size=0.95,
+ api_name="/extract_glb"
+)
+
+client.predict(
+ api_name="/lambda_4"
+)
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..4a3b36f
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1 @@
+gradio_client>=1.7.0
--
GitLab