orpatashnik commited on
Commit
5a477d4
·
1 Parent(s): ad9dc8a
Files changed (2) hide show
  1. app.py +14 -2
  2. example_images/01.jpg +0 -0
app.py CHANGED
@@ -38,7 +38,7 @@ ip_model = NestedAdapterInference(
38
  )
39
 
40
  # Generation defaults
41
- negative_prompt = "monochrome, lowres, bad anatomy, worst quality, low quality"
42
  num_inference_steps = 30
43
  guidance_scale = 5.0
44
 
@@ -77,7 +77,7 @@ def generate_images(img1, img2, img3, prompt, w, num_samples, seed):
77
  # Gradio UI
78
  # ----------------------
79
  with gr.Blocks() as demo:
80
- gr.Markdown("## Personalized Image Generation Demo")
81
  gr.Markdown(
82
  "Upload up to 3 reference images. "
83
  "Faces will be auto-aligned before personalization. Include the placeholder token (e.g., \\<person\\>) in your prompt, "
@@ -95,6 +95,18 @@ with gr.Blocks() as demo:
95
  num_samples_input = gr.Slider(minimum=1, maximum=6, step=1, value=4, label="Number of Images to Generate")
96
  seed_input = gr.Slider(minimum=-1, maximum=100000, step=1, value=-1, label="Random Seed (use -1 for random and up to 100000)")
97
  generate_button = gr.Button("Generate Images")
 
 
 
 
 
 
 
 
 
 
 
 
98
  with gr.Column(scale=1):
99
  output_gallery = gr.Gallery(label="Generated Images", columns=3)
100
 
 
38
  )
39
 
40
  # Generation defaults
41
+ negative_prompt = "bad anatomy, monochrome, lowres, worst quality, low quality"
42
  num_inference_steps = 30
43
  guidance_scale = 5.0
44
 
 
77
  # Gradio UI
78
  # ----------------------
79
  with gr.Blocks() as demo:
80
+ gr.Markdown("## Nested Attention: Semantic-aware Attention Values for Concept Personalization")
81
  gr.Markdown(
82
  "Upload up to 3 reference images. "
83
  "Faces will be auto-aligned before personalization. Include the placeholder token (e.g., \\<person\\>) in your prompt, "
 
95
  num_samples_input = gr.Slider(minimum=1, maximum=6, step=1, value=4, label="Number of Images to Generate")
96
  seed_input = gr.Slider(minimum=-1, maximum=100000, step=1, value=-1, label="Random Seed (use -1 for random and up to 100000)")
97
  generate_button = gr.Button("Generate Images")
98
+
99
+ # Add examples
100
+ gr.Examples(
101
+ examples=[
102
+ ["example_images/01.jpg", None, None, "a pop figure of a <person>, she stands on a white background", 2.0, 4, 1],
103
+ ["example_images/01.jpg", None, None, "a watercolor painting of a <person>, closeup", 1.0, 4, 42],
104
+ ["example_images/01.jpg", None, None, "a high quality photo of a <person> as a firefighter", 3.0, 4, 10],
105
+ ],
106
+ inputs=[img1, img2, img3, prompt_input, w_input, num_samples_input, seed_input],
107
+ label="Example Prompts"
108
+ )
109
+
110
  with gr.Column(scale=1):
111
  output_gallery = gr.Gallery(label="Generated Images", columns=3)
112
 
example_images/01.jpg ADDED