Imane Momayiz
commited on
Commit
•
4821540
1
Parent(s):
eeeb878
test commitscheduler
Browse files
app.py
CHANGED
@@ -8,20 +8,6 @@ from huggingface_hub import HfApi, CommitScheduler
|
|
8 |
import uuid
|
9 |
|
10 |
|
11 |
-
|
12 |
-
def save_feedback(input_text:str, output_1: str, output_2:str, user_choice: int) -> None:
|
13 |
-
"""
|
14 |
-
Append input/outputs and user feedback to a JSON Lines file
|
15 |
-
using a thread lock to avoid concurrent writes from different users.
|
16 |
-
"""
|
17 |
-
with scheduler.lock:
|
18 |
-
with submissions_file.open("a") as f:
|
19 |
-
f.write(json.dumps({"input": input_text, "output_1": output_1, "output_2": output_2, "user_choice": user_choice}))
|
20 |
-
f.write("\n")
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
HF_API_KEY = os.environ.get("HF_TOKEN", None)
|
26 |
|
27 |
api = HfApi(token=HF_API_KEY)
|
@@ -31,13 +17,15 @@ DATASET_REPO_URL = f"https://huggingface.co/datasets/{REPO_ID}"
|
|
31 |
|
32 |
submissions_folder = "submissions"
|
33 |
submissions_file = os.path.join(submissions_folder, f"submissions_{uuid.uuid4()}.json")
|
|
|
34 |
|
35 |
scheduler = CommitScheduler(
|
36 |
repo_id=REPO_ID,
|
37 |
repo_type="dataset",
|
38 |
folder_path=submissions_folder,
|
39 |
path_in_repo="submissions",
|
40 |
-
every=
|
|
|
41 |
)
|
42 |
|
43 |
def load_data(repo_id):
|
|
|
8 |
import uuid
|
9 |
|
10 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
HF_API_KEY = os.environ.get("HF_TOKEN", None)
|
12 |
|
13 |
api = HfApi(token=HF_API_KEY)
|
|
|
17 |
|
18 |
submissions_folder = "submissions"
|
19 |
submissions_file = os.path.join(submissions_folder, f"submissions_{uuid.uuid4()}.json")
|
20 |
+
os.makedirs(submissions_folder, exist_ok=True)
|
21 |
|
22 |
scheduler = CommitScheduler(
|
23 |
repo_id=REPO_ID,
|
24 |
repo_type="dataset",
|
25 |
folder_path=submissions_folder,
|
26 |
path_in_repo="submissions",
|
27 |
+
every=1,
|
28 |
+
hf_api=api
|
29 |
)
|
30 |
|
31 |
def load_data(repo_id):
|