bugfixes

app.py

@@ -8,12 +8,16 @@ import gradio as gr
 import requests
 from transformers import pipeline
 from huggingface_hub import Repository
+from dotenv import load_dotenv
+from pathlib import Path
 
 # These variables are for storing the mturk HITs in a Hugging Face dataset.
+if Path(".env").is_file():
+    load_dotenv(".env")
+DATASET_REPO_URL = os.getenv("DATASET_REPO_URL")
+HF_TOKEN = os.getenv("HF_TOKEN")
 DATA_FILENAME = "data.jsonl"
 DATA_FILE = os.path.join("data", DATA_FILENAME)
-DATASET_REPO_URL = os.environ.get(DATASET_REPO_URL)
-HF_TOKEN = os.environ.get("HF_TOKEN")
 repo = Repository(
     local_dir="data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
 )

collect.py

@@ -20,13 +20,13 @@ parser.add_argument("--live_mode", action="store_true", help="""
 
 args = parser.parse_args()
 
-MTURK_URL = f"https://mturk-requester{"" if args.live_mode else "-sandbox"}.{args.mturk_region}.amazonaws.com"
+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=MTURK_REGION,
+    region_name=args.mturk_region,
     endpoint_url=MTURK_URL,
 )
 
@@ -48,6 +48,6 @@ new_hit = mturk.create_hit(
 )
 
 print(
-    f"Link: https://worker{"" if args.live_mode else "sandbox"}.mturk.com/mturk/preview?groupId="
+    f"Link: https://worker{'' if args.live_mode else 'sandbox'}.mturk.com/mturk/preview?groupId="
     + new_hit["HIT"]["HITGroupId"]
 )

requirements.txt

@@ -3,4 +3,5 @@ torch
 transformers
 gradio
 boto3
-huggingface_hub
+huggingface_hub
+python-dotenv