Spaces:
Runtime error
Runtime error
Added another example
Browse files
app.py
CHANGED
|
@@ -19,22 +19,28 @@ MIN_SIZE = 0.01
|
|
| 19 |
WHITE = 255
|
| 20 |
COLORS = ["red", "blue", "green", "orange", "purple", "turquoise", "olive"]
|
| 21 |
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
|
|
|
| 25 |
EXAMPLE_BOXES = {
|
| 26 |
-
|
| 27 |
[0.35, 0.4, 0.65, 0.9],
|
| 28 |
[0, 0.6, 0.3, 0.9],
|
| 29 |
[0.7, 0.55, 1, 0.85]
|
| 30 |
],
|
| 31 |
-
|
| 32 |
[0.4, 0.45, 0.6, 0.95],
|
| 33 |
[0.2, 0.3, 0.4, 0.85],
|
| 34 |
[0.6, 0.3, 0.8, 0.85],
|
| 35 |
[0.1, 0, 0.9, 0.3]
|
| 36 |
],
|
| 37 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 38 |
[0, 0.5, 0.2, 0.8],
|
| 39 |
[0.2, 0.2, 0.4, 0.5],
|
| 40 |
[0.4, 0.5, 0.6, 0.8],
|
|
@@ -97,7 +103,8 @@ COPY_LINK = """
|
|
| 97 |
<a href="https://huggingface.co/spaces/omer11a/bounded-attention?duplicate=true">
|
| 98 |
<img src="https://bit.ly/3gLdBN6" alt="Duplicate Space">
|
| 99 |
</a>
|
| 100 |
-
Duplicate this space to generate more samples without waiting in queue
|
|
|
|
| 101 |
"""
|
| 102 |
ADVANCED_OPTION_DESCRIPTION = """
|
| 103 |
<div class="tooltip" >Number of guidance steps ⓘ
|
|
@@ -195,7 +202,7 @@ def inference(
|
|
| 195 |
return model(prompts, latents=start_code, guidance_scale=classifier_free_guidance_scale).images
|
| 196 |
|
| 197 |
|
| 198 |
-
@spaces.GPU(duration=
|
| 199 |
def generate(
|
| 200 |
prompt,
|
| 201 |
subject_token_indices,
|
|
@@ -392,19 +399,25 @@ def main():
|
|
| 392 |
#inputs=[prompt, subject_token_indices, filter_token_indices, num_tokens],
|
| 393 |
examples=[
|
| 394 |
[
|
| 395 |
-
|
| 396 |
25, 18, 3, 1, 1,
|
| 397 |
7.5, 1, 5, 0.2, 8,
|
| 398 |
286,
|
| 399 |
],
|
| 400 |
[
|
| 401 |
-
|
| 402 |
18, 12, 3, 1, 1,
|
| 403 |
7.5, 1, 5, 0.2, 8,
|
| 404 |
216,
|
| 405 |
],
|
| 406 |
[
|
| 407 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 408 |
18, 12, 3, 1, 1,
|
| 409 |
7.5, 1, 5, 0.2, 8,
|
| 410 |
156,
|
|
|
|
| 19 |
WHITE = 255
|
| 20 |
COLORS = ["red", "blue", "green", "orange", "purple", "turquoise", "olive"]
|
| 21 |
|
| 22 |
+
PROMPT2 = "3 D Pixar animation of a cute unicorn and a pink hedgehog and a nerdy owl traveling in a magical forest"
|
| 23 |
+
PROMPT3 = "science fiction movie poster with an astronaut and a robot and a green alien and a spaceship"
|
| 24 |
+
PROMPT4 = "a realistic photo of a highway with a semi trailer and a concrete mixer and a helicopter"
|
| 25 |
+
PROMPT5 = "a golden retriever and a german shepherd and a boston terrier and an english bulldog and a border collie in a pool"
|
| 26 |
EXAMPLE_BOXES = {
|
| 27 |
+
PROMPT2 : [
|
| 28 |
[0.35, 0.4, 0.65, 0.9],
|
| 29 |
[0, 0.6, 0.3, 0.9],
|
| 30 |
[0.7, 0.55, 1, 0.85]
|
| 31 |
],
|
| 32 |
+
PROMPT3: [
|
| 33 |
[0.4, 0.45, 0.6, 0.95],
|
| 34 |
[0.2, 0.3, 0.4, 0.85],
|
| 35 |
[0.6, 0.3, 0.8, 0.85],
|
| 36 |
[0.1, 0, 0.9, 0.3]
|
| 37 |
],
|
| 38 |
+
PROMPT4: [
|
| 39 |
+
[0.05, 0.5, 0.45, 0.85],
|
| 40 |
+
[0.55, 0.6, 0.95, 0.85],
|
| 41 |
+
[0.3, 0.2, 0.7, 0.45],
|
| 42 |
+
],
|
| 43 |
+
PROMPT5: [
|
| 44 |
[0, 0.5, 0.2, 0.8],
|
| 45 |
[0.2, 0.2, 0.4, 0.5],
|
| 46 |
[0.4, 0.5, 0.6, 0.8],
|
|
|
|
| 103 |
<a href="https://huggingface.co/spaces/omer11a/bounded-attention?duplicate=true">
|
| 104 |
<img src="https://bit.ly/3gLdBN6" alt="Duplicate Space">
|
| 105 |
</a>
|
| 106 |
+
Duplicate this space to generate more samples without waiting in queue.
|
| 107 |
+
To get better results, use our code on your own GPU and increase the number of guidance steps to 15.
|
| 108 |
"""
|
| 109 |
ADVANCED_OPTION_DESCRIPTION = """
|
| 110 |
<div class="tooltip" >Number of guidance steps ⓘ
|
|
|
|
| 202 |
return model(prompts, latents=start_code, guidance_scale=classifier_free_guidance_scale).images
|
| 203 |
|
| 204 |
|
| 205 |
+
@spaces.GPU(duration=340)
|
| 206 |
def generate(
|
| 207 |
prompt,
|
| 208 |
subject_token_indices,
|
|
|
|
| 399 |
#inputs=[prompt, subject_token_indices, filter_token_indices, num_tokens],
|
| 400 |
examples=[
|
| 401 |
[
|
| 402 |
+
PROMPT2, "7,8,17;11,12,17;15,16,17", "5,6,9,10,13,14,18,19", "21",
|
| 403 |
25, 18, 3, 1, 1,
|
| 404 |
7.5, 1, 5, 0.2, 8,
|
| 405 |
286,
|
| 406 |
],
|
| 407 |
[
|
| 408 |
+
PROMPT3, "7;10;13,14;17", "5,6,8,9,11,12,15,16", "17",
|
| 409 |
18, 12, 3, 1, 1,
|
| 410 |
7.5, 1, 5, 0.2, 8,
|
| 411 |
216,
|
| 412 |
],
|
| 413 |
[
|
| 414 |
+
PROMPT4, "9,10;13,14;17", "1,4,5,7,8,11,12,15,16", "17",
|
| 415 |
+
25, 18, 3, 1, 1,
|
| 416 |
+
7.5, 1, 5, 0.2, 8,
|
| 417 |
+
86,
|
| 418 |
+
],
|
| 419 |
+
[
|
| 420 |
+
PROMPT5, "2,3;6,7;10,11;14,15;18,19", "1,4,5,8,9,12,13,16,17,20,21", "22",
|
| 421 |
18, 12, 3, 1, 1,
|
| 422 |
7.5, 1, 5, 0.2, 8,
|
| 423 |
156,
|