lalalala

Dockerfile

@@ -41,9 +41,13 @@ WORKDIR /data
 RUN chmod +x train.sh
 RUN chmod +x upload_results.py
 
+# Expose the secret so it can be used later.
+RUN --mount=type=secret,id=DATACOMP_TOKEN
+#,mode=0444,required=true \
+
 # Could also use CMD. Example:
 # CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
-ENTRYPOINT ["/train.sh:app"]
+ENTRYPOINT ["app:/train.sh"]
 
 
 

upload_results.py

@@ -15,7 +15,7 @@ parser.add_argument(
 )
 parser.add_argument(
     "--model_out_dir",
-    default="teaching_arithmetic/out/addition_plan/",
+    default="teaching_arithmetic/out/addition_plain/",
     type=str,
     required=False,
     help="Path to the model output directory.",