Add examples
Browse files- x2rgb/gradio_demo_x2rgb.py +31 -0
x2rgb/gradio_demo_x2rgb.py
CHANGED
@@ -182,6 +182,37 @@ with gr.Blocks() as demo:
|
|
182 |
elem_id="gallery",
|
183 |
columns=2,
|
184 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
185 |
|
186 |
run_button.click(
|
187 |
fn=generate,
|
|
|
182 |
elem_id="gallery",
|
183 |
columns=2,
|
184 |
)
|
185 |
+
examples = gr.Examples(
|
186 |
+
examples=[
|
187 |
+
"x2rgb/example/kitchen-albedo.png", # albedo
|
188 |
+
"x2rgb/example/kitchen-normal.png", # normal
|
189 |
+
"x2rgb/example/kitchen-roughness.png", # roughness
|
190 |
+
"x2rgb/example/kitchen-metallic.png", # metallic
|
191 |
+
"x2rgb/example/kitchen-irradiance.png", # irradiance
|
192 |
+
"Modern Kitchen", # prompt
|
193 |
+
0, # seed
|
194 |
+
50, # inference_step
|
195 |
+
1, # num_samples
|
196 |
+
7.5, # guidance_scale
|
197 |
+
1.5, # image_guidance_scale
|
198 |
+
],
|
199 |
+
inputs=[
|
200 |
+
albedo,
|
201 |
+
normal,
|
202 |
+
roughness,
|
203 |
+
metallic,
|
204 |
+
irradiance,
|
205 |
+
prompt,
|
206 |
+
seed,
|
207 |
+
inference_step,
|
208 |
+
num_samples,
|
209 |
+
guidance_scale,
|
210 |
+
image_guidance_scale,
|
211 |
+
],
|
212 |
+
outputs=[result_gallery],
|
213 |
+
# cache_mode="eager",
|
214 |
+
# cache_examples=True
|
215 |
+
)
|
216 |
|
217 |
run_button.click(
|
218 |
fn=generate,
|