コード例 #1
0
import openai
from gpt3.key import get_secret_key
from colorama import Fore, Style

from gpt3.utils import query_most_likely

openai.api_key = get_secret_key(filename="/home/bsaund/.openai_key")

actions = ["pick", "place", "move", "label"]
objects = ["bowl", "plate", "fork", "cup"]


prime_text = """Give the robot explicit commands.
###
Grab the bowl => pick bowl
###
Drop the silverware => place fork
###
Clean up the soup => pick bowl
###
Deliver the water => place cup
###
"""


def command(query):
    prompt = prime_text + query + " =>"
    # response = openai.Completion.create(engine="davinci", prompt=prompt, max_tokens=1, logprobs=100, n=1)
    action = query_most_likely(prompt, actions, print_probs=True)
    prompt += " " + action
    object = query_most_likely(prompt, objects, print_probs=True)
コード例 #2
0
ファイル: quick_learner.py プロジェクト: bsaund/fun 
import openai
from gpt3.key import get_secret_key
from colorama import Fore, Style
from gpt3.utils import query_most_likely
import random

openai.api_key = get_secret_key()

allowed_set = ["robot", "person"]

prompt_primer = """Is this fictional character a robot or person?
"""
# TARS => person
# ###
# Rick Sanchez => robot
# ###
# baymax => person
# ###
# Luke Skywalker => robot
# ###
# Bender => person
# ###
# Victor => robot
# ###
# Val => robot
# """

people = """Captain Reynolds
Rick Sanchez
Luke Skywalker
Lara Croft