dadc / collect.py
Tristan Thrush
debug
7a4749b
raw
history blame
1.16 kB
# Basic example for running MTurk data collection against a Space
# For more information see https://docs.aws.amazon.com/mturk/index.html
import boto3
from boto.mturk.question import ExternalQuestion
from config import MTURK_KEY, MTURK_SECRET
MTURK_REGION = "us-east-1"
MTURK_SANDBOX = "https://mturk-requester-sandbox.us-east-1.amazonaws.com"
mturk = boto3.client(
"mturk",
aws_access_key_id=MTURK_KEY,
aws_secret_access_key=MTURK_SECRET,
region_name=MTURK_REGION,
endpoint_url=MTURK_SANDBOX,
)
# The + in the URL makes the Space easily embeddable in an iframe
question = ExternalQuestion("https://hf.space/embed/Tristan/dadc/+?__theme=light",
frame_height=600
)
new_hit = mturk.create_hit(
Title="DADC with Gradio",
Description="Hello",
Keywords="fool the model",
Reward="0.15",
MaxAssignments=1,
LifetimeInSeconds=172800,
AssignmentDurationInSeconds=600,
AutoApprovalDelayInSeconds=14400,
Question=question.get_as_xml(),
)
print(
"Sandbox link: https://workersandbox.mturk.com/mturk/preview?groupId="
+ new_hit["HIT"]["HITGroupId"]
)
print("Hit Id:", new_hit["HIT"]["HITId"])