curryporkchop commited on
Commit
0436f12
·
1 Parent(s): 0e6fe1d

Add initial project files for TextTo3D app

Browse files
Files changed (4) hide show
  1. README.md +2 -3
  2. app.py +100 -0
  3. gitattributes +35 -0
  4. requirements.txt +2 -0
README.md CHANGED
@@ -1,13 +1,12 @@
1
  ---
2
  title: TextTo3D 2.0
3
- emoji: 💻
4
  colorFrom: purple
5
- colorTo: blue
6
  sdk: gradio
7
  sdk_version: 5.17.0
8
  app_file: app.py
9
  pinned: false
10
- short_description: CTP 2025 CLASS DEMO
11
  ---
12
 
13
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
1
  ---
2
  title: TextTo3D 2.0
3
+ emoji: 🐨
4
  colorFrom: purple
5
+ colorTo: purple
6
  sdk: gradio
7
  sdk_version: 5.17.0
8
  app_file: app.py
9
  pinned: false
 
10
  ---
11
 
12
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
app.py ADDED
@@ -0,0 +1,100 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from gradio_client import Client, handle_file
3
+
4
+ # Create a client for the external 3D generation API hosted on "tencent/Hunyuan3D-2"
5
+ api_client = Client("tencent/Hunyuan3D-2")
6
+
7
+ def generate_3d(caption: str, image, steps: float, guidance_scale: float, seed: float, octree_resolution: str, check_box_rembg: bool):
8
+ """
9
+ Calls the external /generation_all API endpoint to generate 3D outputs.
10
+ Returns a tuple of:
11
+ [0] White Mesh file (filepath),
12
+ [1] Textured Mesh file (filepath),
13
+ [2] HTML output string,
14
+ [3] Additional HTML output string.
15
+ """
16
+ # If no image is provided, use a default image URL
17
+ if image is None:
18
+ image = "https://raw.githubusercontent.com/gradio-app/gradio/main/test/test_files/bus.png"
19
+
20
+ image_path = handle_file(image)
21
+
22
+ result = api_client.predict(
23
+ caption=caption,
24
+ image=image_path,
25
+ steps=steps,
26
+ guidance_scale=guidance_scale,
27
+ seed=seed,
28
+ octree_resolution=octree_resolution,
29
+ check_box_rembg=check_box_rembg,
30
+ api_name="/generation_all"
31
+ )
32
+ return result
33
+
34
+ # Build the UI using Gradio Blocks for a more organized layout.
35
+ with gr.Blocks(title="3D Generator") as demo:
36
+ gr.Markdown("# 3D Generator")
37
+ gr.Markdown("Generate 3D models using the external `/generation_all` API from `tencent/Hunyuan3D-2`.")
38
+
39
+ with gr.Row():
40
+ with gr.Column():
41
+ caption_input = gr.Textbox(
42
+ label="Text Prompt (Caption)",
43
+ placeholder="Enter your caption here",
44
+ lines=2
45
+ )
46
+ image_input = gr.Image(
47
+ label="Input Image",
48
+ type="filepath"
49
+ )
50
+ with gr.Column():
51
+ steps_input = gr.Slider(
52
+ minimum=1, maximum=100, step=1, value=50,
53
+ label="Inference Steps"
54
+ )
55
+ guidance_scale_input = gr.Slider(
56
+ minimum=1.0, maximum=10.0, step=0.1, value=5.5,
57
+ label="Guidance Scale"
58
+ )
59
+ seed_input = gr.Slider(
60
+ minimum=0, maximum=10000, step=1, value=1234,
61
+ label="Seed"
62
+ )
63
+ octree_resolution_input = gr.Dropdown(
64
+ choices=["256", "384", "512"],
65
+ value="256",
66
+ label="Octree Resolution"
67
+ )
68
+ check_box_rembg_input = gr.Checkbox(
69
+ label="Remove Background",
70
+ value=True
71
+ )
72
+
73
+ generate_button = gr.Button("Generate 3D Model")
74
+
75
+ gr.Markdown("### Generated Outputs")
76
+ with gr.Row():
77
+ white_mesh_output = gr.File(label="Download White Mesh")
78
+ textured_mesh_output = gr.File(label="Download Textured Mesh")
79
+ output_html1 = gr.HTML(label="Output HTML 1")
80
+ output_html2 = gr.HTML(label="Output HTML 2")
81
+
82
+ # Connect the button click event to the generate_3d function.
83
+ generate_button.click(
84
+ fn=generate_3d,
85
+ inputs=[
86
+ caption_input, image_input, steps_input,
87
+ guidance_scale_input, seed_input,
88
+ octree_resolution_input, check_box_rembg_input
89
+ ],
90
+ outputs=[
91
+ white_mesh_output, textured_mesh_output,
92
+ output_html1, output_html2
93
+ ]
94
+ )
95
+
96
+ gr.Markdown("## Enjoy your generated 3D models!")
97
+
98
+ if __name__ == "__main__":
99
+ # Launch the app with a public share link (if desired)
100
+ demo.launch(share=True)
gitattributes ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ gradio
2
+ gradio_client