update dockerfile

load_data.py

@@ -11,7 +11,7 @@ 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 = os.getenv("API_KEY")
     rg.init(api_key=api_key, workspace="admin")
 
     # load dataset from json
@@ -102,6 +102,9 @@ def load_datasets():
 
 
 while True:
+    print("Loading...")
+    api_key = os.getenv("API_KEY")
+    print("API KEY", api_key)
     try:
         response = requests.get("http://0.0.0.0:6900/")
         if response.status_code == 200: