fix permission

Dockerfile

@@ -9,8 +9,9 @@ WORKDIR /code
 COPY ./requirements.txt /code/requirements.txt
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
-RUN mkdir /code/tmp
 
 COPY . .
 
+USER root
+
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

app/watermelon.py

@@ -35,13 +35,11 @@ def submit(response: Response):
     timestamp = beijing().strftime("%Y%m%d_%H%M%S_%f")
     filename = timestamp+".jsonl"
     marker = timestamp+".submit"
-    filepath_remote = "/".join([game, filename])
-    markerpath_remote = "/".join([game, marker])
-    filepath_local = "/".join(["tmp", filename])
-    markerpath_local = "/".join(["tmp", marker])
+    filepath = "/".join([game, filename])
+    markerpath = "/".join([game, marker])
     if not os.path.exists("tmp"):
         os.mkdir("tmp")
-    with open(filepath_local, "w") as f:
+    with open(filepath, "w") as f:
         for key in train_generated:
             myobj = {
                 "command": key,
@@ -50,18 +48,18 @@ def submit(response: Response):
             }
             json.dump(myobj, f, indent=None)
             f.write("\n")
-    open(markerpath_local, "w").close()
+    open(markerpath, "w").close()
     db_token = os.environ.get("db_token")
     try:
         hfapi.upload_file(
-            path_or_fileobj=filepath_local,
-            path_in_repo=filepath_remote,
+            path_or_fileobj=filepath,
+            path_in_repo=filepath,
             repo_id="pgsoft/features",
             repo_type="dataset",
             token=db_token)
         hfapi.upload_file(
-            path_or_fileobj=markerpath_local,
-            path_in_repo=markerpath_remote,
+            path_or_fileobj=markerpath,
+            path_in_repo=markerpath,
             repo_id="pgsoft/features",
             repo_type="dataset",
             token=db_token)