OBJECT = { "backpack": "backpack", "backpack_dog": "backpack", "bear_plushie": "stuffed animal", "berry_bowl": "bowl", "can": "can", "candle": "candle", "clock": "clock", "colorful_sneaker": "sneaker", "duck_toy": "toy", "fancy_boot": "boot", "grey_sloth_plushie": "stuffed animal", "monster_toy": "toy", "pink_sunglasses": "glasses", "poop_emoji": "toy", "rc_car": "toy", "red_cartoon": "cartoon", "robot_toy": "toy", "shiny_sneaker": "sneaker", "teapot": "teapot", "vase": "vase", "wolf_plushie": "stuffed animal" } LIVE_OBJECT = { "cat": "cat", "cat2": "cat", "dog": "dog", "dog2": "dog", "dog3": "dog", "dog5": "dog", "dog6": "dog", "dog7": "dog", "dog8": "dog", } OBJECT_PROMPTS = [ 'a {} in the jungle', 'a {} in the snow', 'a {} on the beach', 'a {} on a cobblestone street', 'a {} on top of pink fabric', 'a {} on top of a wooden floor', 'a {} with a city in the background', 'a {} with a mountain in the background', 'a {} with a blue house in the background', 'a {} on top of a purple rug in a forest', 'a {} with a wheat field in the background', 'a {} with a tree and autumn leaves in the background', 'a {} with the Eiffel Tower in the background', 'a {} floating on top of water', 'a {} floating in an ocean of milk', 'a {} on top of green grass with sunflowers around it', 'a {} on top of a mirror', 'a {} on top of the sidewalk in a crowded street', 'a {} on top of a dirt road', 'a {} on top of a white rug', 'a red {}', 'a purple {}', 'a shiny {}', 'a wet {}', 'a cube shaped {}' ] LIVE_OBJECT_PROMPTS = [ 'a {} in the jungle', 'a {} in the snow', 'a {} on the beach', 'a {} on a cobblestone street', 'a {} on top of pink fabric', 'a {} on top of a wooden floor', 'a {} with a city in the background', 'a {} with a mountain in the background', 'a {} with a blue house in the background', 'a {} on top of a purple rug in a forest', 'a {} wearing a red hat', 'a {} wearing a santa hat', 'a {} wearing a rainbow scarf', 'a {} wearing a black top hat and a monocle', 'a {} in a chef outfit', 'a {} in a firefighter outfit', 'a {} in a police outfit', 'a {} wearing pink glasses', 'a {} wearing a yellow shirt', 'a {} in a purple wizard outfit', 'a red {}', 'a purple {}', 'a shiny {}', 'a wet {}', 'a cube shaped {}' ] KOSMOSG_OBJECT_PROMPTS = [ '{} in the jungle', '{} in the snow', '{} on the beach', '{} on a cobblestone street', '{} on top of pink fabric', '{} on top of a wooden floor', '{} with a city in the background', '{} with a mountain in the background', '{} with a blue house in the background', '{} on top of a purple rug in a forest', '{} with a wheat field in the background', '{} with a tree and autumn leaves in the background', '{} with the Eiffel Tower in the background', '{} floating on top of water', '{} floating in an ocean of milk', '{} on top of green grass with sunflowers around it', '{} on top of a mirror', '{} on top of the sidewalk in a crowded street', '{} on top of a dirt road', '{} on top of a white rug', '{}, red', '{}, purple', '{}, shiny', '{}, wet', '{}, cube shaped' ] KOSMOSG_LIVE_OBJECT_PROMPTS = [ '{} in the jungle', '{} in the snow', '{} on the beach', '{} on a cobblestone street', '{} on top of pink fabric', '{} on top of a wooden floor', '{} with a city in the background', '{} with a mountain in the background', '{} with a blue house in the background', '{} on top of a purple rug in a forest', '{} wearing a red hat', '{} wearing a santa hat', '{} wearing a rainbow scarf', '{} wearing a black top hat and a monocle', '{} in a chef outfit', '{} in a firefighter outfit', '{} in a police outfit', '{} wearing pink glasses', '{} wearing a yellow shirt', '{} in a purple wizard outfit', '{}, red', '{}, purple', '{}, shiny', '{}, wet', '{}, cube shaped' ]