Spaces:
Running
on
Zero
Running
on
Zero
initial ui changes
Browse filesadd random seed + advanced options
app.py
CHANGED
@@ -37,6 +37,13 @@ CONCEPTS_MAP = {'age':'age','animal fur':'animal_fur', 'deterioration': 'deterio
|
|
37 |
|
38 |
concept_options = list(CONCEPTS_MAP.keys())
|
39 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
40 |
@spaces.GPU
|
41 |
def get_image_embeds(pil_image, model=clip_model, preproc=preprocess, dev=device):
|
42 |
"""Get CLIP image embeddings for a given PIL image"""
|
@@ -188,19 +195,25 @@ with gr.Blocks(title="Image Concept Composition") as demo:
|
|
188 |
concept_name3 = gr.Dropdown(concept_options, label="concept 3", value= None, info="concept type")
|
189 |
rank3 = gr.Slider(minimum=1, maximum=50, value=30, step=1, label="Rank 3")
|
190 |
|
191 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
192 |
|
193 |
-
with gr.Row():
|
194 |
-
scale = gr.Slider(minimum=0.1, maximum=2.0, value=1.0, step=0.1, label="Scale")
|
195 |
-
seed = gr.Number(value=420, label="Seed", precision=0)
|
196 |
|
197 |
-
submit_btn = gr.Button("Generate")
|
198 |
|
199 |
with gr.Column():
|
200 |
output_image = gr.Image(label="composed output", show_label=True)
|
201 |
|
202 |
submit_btn.click(
|
203 |
-
fn=
|
|
|
|
|
|
|
204 |
inputs=[
|
205 |
base_image,
|
206 |
concept_image1, concept_name1,
|
|
|
37 |
|
38 |
concept_options = list(CONCEPTS_MAP.keys())
|
39 |
|
40 |
+
|
41 |
+
MAX_SEED = np.iinfo(np.int32).max
|
42 |
+
def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
43 |
+
if randomize_seed:
|
44 |
+
seed = random.randint(0, MAX_SEED)
|
45 |
+
return seed
|
46 |
+
|
47 |
@spaces.GPU
|
48 |
def get_image_embeds(pil_image, model=clip_model, preproc=preprocess, dev=device):
|
49 |
"""Get CLIP image embeddings for a given PIL image"""
|
|
|
195 |
concept_name3 = gr.Dropdown(concept_options, label="concept 3", value= None, info="concept type")
|
196 |
rank3 = gr.Slider(minimum=1, maximum=50, value=30, step=1, label="Rank 3")
|
197 |
|
198 |
+
submit_btn = gr.Button("Generate")
|
199 |
+
|
200 |
+
with gr.Accordion("Advanced options", open=False):
|
201 |
+
prompt = gr.Textbox(label="Guidance Prompt (Optional)", placeholder="Optional text prompt to guide generation")
|
202 |
+
with gr.Row():
|
203 |
+
scale = gr.Slider(minimum=0.1, maximum=2.0, value=1.0, step=0.1, label="Scale")
|
204 |
+
randomize_seed = gr,Checkbox(value=True, label="Randomize seed")
|
205 |
+
seed = gr.Number(value=0, label="Seed", precision=0)
|
206 |
|
|
|
|
|
|
|
207 |
|
|
|
208 |
|
209 |
with gr.Column():
|
210 |
output_image = gr.Image(label="composed output", show_label=True)
|
211 |
|
212 |
submit_btn.click(
|
213 |
+
fn=randomize_seed_fn,
|
214 |
+
inputs=[seed, randomize_seed],
|
215 |
+
outputs=seed,
|
216 |
+
).then(fn=process_and_display,
|
217 |
inputs=[
|
218 |
base_image,
|
219 |
concept_image1, concept_name1,
|