Update app.py

app.py

@@ -116,9 +116,29 @@ QUESTION_SET = [
 def load_or_initialize_count_json(audio_paths):
     # Add filelock to /workspace/count.json
     lock_path = COUNT_JSON_PATH + ".lock"
+
+    try:
+        api = HfApi()
+        repo_id = "intersteller2887/Turing-test-dataset"
+        hf_token = os.getenv("HF_TOKEN")
+        count_json = None
+
+        try:
+            count_json = api.get_file(
+                path_in_repo=COUNT_JSON_REPO_PATH,
+                repo_id=repo_id,
+                repo_type="dataset",
+                token=hf_token
+            )
+
+    except Exception as e:
+        print(f"Could not download count.json from dataset {e}")
+
     # Read of count.json will wait for 10 seconds until another thread involving releases it, and then add a lock to it
     with FileLock(lock_path, timeout=10):
         # If count.json exists: load into count_data
+        with open(COUNT_JSON_PATH, "wb") as f:
+            f.write(count_json)
         # Else initialize count_data with orderedDict
         if os.path.exists(COUNT_JSON_PATH):
             with open(COUNT_JSON_PATH, "r", encoding="utf-8") as f: