update dockerfile

load_data.py

@@ -1,5 +1,5 @@
 import os
-
+import sys
 import requests
 import time
 import pandas as pd
@@ -11,7 +11,8 @@ from argilla.labeling.text_classification import Rule, add_rules
 def load_datasets():
     # This is the code that you want to execute when the endpoint is available
     print("Argilla is available! Loading datasets")
-    api_key = os.getenv("API_KEY")
+    api_key = sys.argv[-1]
+    print(api_key)
     rg.init(api_key=api_key, workspace="admin")
 
     # load dataset from json

start.sh

@@ -38,7 +38,7 @@ sudo sed -i 's,ARGILLA_PASSWORD,'"$ARGILLA_PASSWORD"',g' /packages/users.yml
 # Load data
 echo "Loading data"
 pip3 install datasets
-python3.9 /load_data.py &
+python3.9 /load_data.py "$API_KEY" &
 
 # Start argilla
 echo "start argilla"