blanchon commited on
Commit
61639d4
·
1 Parent(s): 83a5122

Add examples

Browse files
Files changed (1) hide show
  1. 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,