stoc

import random

from transformers import AutoModelForPrompt, AutoTokenizer from PIL import Image

Load the model and tokenizer

model = AutoModelForPrompt.from_pretrained("bigscience/bigsleep-14m") tokenizer = AutoTokenizer.from_pretrained("bigscience/bigsleep-14m")

Define a list of Stranger Things character names

names = ["Mike", "Eleven", "Dustin", "Lucas", "Will", "Joyce", "Jim", "Nancy", "Jonathan", "Steve", "Robin", "Erica", "Max", "Billy"]

Define a list of Stranger Things character traits

traits = ["brave", "intelligent", "witty", "sarcastic", "determined", "loyal", "protective", "kind", "compassionate", "troubled", "mysterious", "powerful"]

Define a list of Stranger Things locations

locations = ["Hawkins National Laboratory", "Mirkwood Forest", "The Upside Down", "Starcourt Mall", "Hawkins Middle School", "Hawkins High School"]

Generate a random prompt for the model

prompt = f"A photorealistic portrait of a Stranger Things OC named {random.choice(names)} with {random.choice(traits)} eyes and {random.choice(traits)} hair. The OC is standing in {random.choice(locations)}."

#Generate the image image = model.generate_image(prompt)

#Save the image to a file image.save("oc.png")

#Display the image Image.open("oc.png").show()