|
|
|
|
|
|
|
import boto3 |
|
from boto.mturk.question import ExternalQuestion |
|
|
|
from config import MTURK_KEY, MTURK_SECRET |
|
import argparse |
|
|
|
parser = argparse.ArgumentParser() |
|
parser.add_argument("--mturk_region", default="us-east-1", help="The region for mturk (default: us-east-1)") |
|
parser.add_argument("--space_name", default="Tristan/dadc", help="Name of the accompanying Hugging Face space (default: Tristan/dadc)") |
|
parser.add_argument("--num_hits", type=int, default=5, help="The number of HITs.") |
|
parser.add_argument("--num_assignments", type=int, default=1, help="The number of times that the HIT can be accepted and completed.") |
|
parser.add_argument("--live_mode", action="store_true", help=""" |
|
Whether to run in live mode with real turkers. This will charge your account money. |
|
If you don't use this flag, the HITs will be deployed on the sandbox version of mturk, |
|
which will not charge your account money. |
|
""" |
|
) |
|
|
|
args = parser.parse_args() |
|
|
|
MTURK_URL = f"https://mturk-requester{'' if args.live_mode else '-sandbox'}.{args.mturk_region}.amazonaws.com" |
|
|
|
mturk = boto3.client( |
|
"mturk", |
|
aws_access_key_id=MTURK_KEY, |
|
aws_secret_access_key=MTURK_SECRET, |
|
region_name=args.mturk_region, |
|
endpoint_url=MTURK_URL, |
|
) |
|
|
|
|
|
question = ExternalQuestion(f"https://hf.space/embed/{args.space_name}/+?__theme=light", |
|
frame_height=600 |
|
) |
|
|
|
for i in range(args.num_hits): |
|
new_hit = mturk.create_hit( |
|
Title="Beat the AI", |
|
Description="Try to fool an AI by creating examples that it gets wrong", |
|
Keywords="fool the model", |
|
Reward="0.15", |
|
MaxAssignments=args.num_assignments, |
|
LifetimeInSeconds=172800, |
|
AssignmentDurationInSeconds=600, |
|
AutoApprovalDelayInSeconds=14400, |
|
Question=question.get_as_xml(), |
|
) |
|
|
|
print( |
|
f"HIT Group Link: https://worker{'' if args.live_mode else 'sandbox'}.mturk.com/mturk/preview?groupId=" |
|
+ new_hit["HIT"]["HITGroupId"] |
|
) |
|
|
|
|
