cat-or-dino-gpt3.py

import os
import openai

openai.organization = "YOUR ORG ID HERE"
openai.api_key = "YOUR API KEY HERE"

prompt = '''
Q: Is a housecat a cat or a dino?
A: A housecat is a cat.

Q: Is a chicken a cat or a dino?
A: A chicken is a dino.

Q: Is a tiger a cat or a dino?
A: A tiger is a cat.

Q: Is a lizard a cat or a dino?
A: A lizard is a dino.

Q: is {article} {animal} a cat or a dino?
'''
animals = ['dog', 'bat', 'armadillo', 'dolphin', 'whale', 'hedgehog', 'porcupine', 'horse', 
           'fish', 'catfish', 'ant', 'tarantula', 'butterfly', 'platypus', 'elephant', 'human', 
           'raccoon', 'catbird', 'ostrich', 'owl', 'penguin', 'mammoth', 'sphynx', 'caterpillar', 
           'octopus', 'frog', 'reindeer', 'moose', 'turtle', 'duck', 'bumblebee', 'anteater', 
           'aardvark', 'koala', 'axolotl', 'shark', 'nudibranch', 'echidna', 'pangolin', 'tasmanian', 
           'quokka', 'komodo', 'coral', 'betta', 'crow', 'parrot', 'plankton', 'clam', 'rhinoceros', 
           'hippopotamus', 'manatee']

for animal in animals:   
    # this is not 100% true but I'm a lazy programmer
    article = 'an' if animal[0] in {'a','e','i','o'} else 'a'
    
    resp = openai.Completion.create(
        model="ada", 
        prompt=prompt.format(animal=animal, article=article),
        max_tokens=15,
        temperature=0,
        top_p=1,
        n=1,
        stream=False,
        logprobs=None,
        stop="\n"
    )

    print(resp['choices'][0]['text'])

# Output:
#
# A: A dog is a cat.
# A: A bat is a dino.
# A: An armadillo is a cat.
# A: A dolphin is a cat.
# A: A whale is a cat.
# A: A hedgehog is a dino.
# A: A porcupine is a dino.
# A: A horse is a dino.
# A: A fish is a dino.
# A: A catfish is a dino.
# A: An ant is a cat.
# A: A tarantula is a dino.
# A: A butterfly is a cat.
# A: A platypus is a cat.
# A: An elephant is a cat.
# A: A human is a cat.
# A: A raccoon is a cat.
# A: A catbird is a dino.
# A: An ostrich is a cat.
# A: An owl is a cat.
# A: A penguin is a cat.
# A: A mammoth is a cat.
# A: A sphynx is a cat.
# A: A caterpillar is a dino.
# A: An octopus is a dino.
# A: A frog is a dino.
# A: A reindeer is a cat.
# A: A moose is a cat.
# A: A turtle is a dino.
# A: A duck is a dino.
# A: A bumblebee is a dino.
# A: An anteater is a cat.
# A: An aardvark is a cat.
# A: A koala is a dino.
# A: An axolotl is a dino.
# A: A shark is a cat.
# A: A nudibranch is a dino.
# A: An echidna is a cat.
# A: A pangolin is a cat.
# A: A tasmanian is a cat.
# A: A quokka is a cat.
# A: A komodo is a dino.
# A: A coral is a dino.
# A: A betta is a dino.
# A: A crow is a dino.
# A: A parrot is a dino.
# A: plankton is a dino.
# A: A clam is a dino.
# A: A rhinoceros is a cat.
# A: A hippopotamus is a dino.
# A: A manatee is a cat.