linoyts HF Staff commited on
Commit
8bb102d
·
verified ·
1 Parent(s): c091a06

add examples

Browse files
Files changed (1) hide show
  1. app.py +33 -0
app.py CHANGED
@@ -69,6 +69,27 @@ RANKS_MAP={
69
  concept_options = list(CONCEPTS_MAP.keys())
70
 
71
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
72
  MAX_SEED = np.iinfo(np.int32).max
73
  def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
74
  if randomize_seed:
@@ -248,6 +269,18 @@ following the algorithm proposed in [*IP-Composer: Semantic Composition of Visua
248
  output_image = gr.Image(label="composed output", show_label=True)
249
  submit_btn = gr.Button("Generate")
250
 
 
 
 
 
 
 
 
 
 
 
 
 
251
 
252
  submit_btn.click(
253
  fn=randomize_seed_fn,
 
69
  concept_options = list(CONCEPTS_MAP.keys())
70
 
71
 
72
+ examples = [
73
+ ['./IP_Composer/assets/objects/mug.png', './IP_Composer/assets/patterns/splash.png', 'patterns (including color)', None, None, None, None, 80, 30, 30, None,1.0,0]
74
+
75
+ ]
76
+
77
+ def generate_examples(base_image,
78
+ concept_image1, concept_name1,
79
+ concept_image2, concept_name2,
80
+ concept_image3, concept_name3,
81
+ rank1, rank2, rank3,
82
+ prompt, scale, seed):
83
+ return process_and_display(base_image,
84
+ concept_image1, concept_name1,
85
+ concept_image2, concept_name2,
86
+ concept_image3, concept_name3,
87
+ rank1, rank2, rank3,
88
+ prompt, scale, seed)
89
+
90
+
91
+
92
+
93
  MAX_SEED = np.iinfo(np.int32).max
94
  def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
95
  if randomize_seed:
 
269
  output_image = gr.Image(label="composed output", show_label=True)
270
  submit_btn = gr.Button("Generate")
271
 
272
+ gr.Examples(
273
+ examples,
274
+ inputs=[base_image,
275
+ concept_image1, concept_name1,
276
+ concept_image2, concept_name2,
277
+ concept_image3, concept_name3,
278
+ rank1, rank2, rank3,
279
+ prompt, scale, seed],
280
+ outputs=[output_image],
281
+ fn=generate_examples,
282
+ cache_examples=False
283
+ )
284
 
285
  submit_btn.click(
286
  fn=randomize_seed_fn,