update dockerfile

load_data.py

@@ -7,6 +7,8 @@ def load_data():
     import argilla as rg
     from datasets import load_dataset
 
+    rg.init(api_key="admin.apikey")
+
     # load dataset from the hub
     dataset = load_dataset("argilla/gutenberg_spacy-ner", split="train")
 
@@ -20,13 +22,14 @@ def load_data():
 if __name__ == '__main__':
     while True:
         try:
-            response = requests.get("http://0.0.0.0:6900/")
+            response = requests.get("http://localhost:6900/")
             if response.status_code == 200:
                 load_data()
                 print("Data Loaded!")
                 break
             else:
                 time.sleep(5)
+
         except Exception as e:
             print(e)
-            pass
+            time.sleep(10)