region:us
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()
