Log straight away

app.py

@@ -47,9 +47,12 @@ if getenv("HF_API_TOKEN") is not None:
     #log_queue = Queue()
     #t = Thread(target=check_thread, args=(log_queue,))
     #t.start()
-
-    # handle zeroGPU
-    log_queue = None
+    
+    logging_callback = log_data(
+        hf_token=getenv("HF_API_TOKEN"),
+        dataset_name=getenv("OUTPUT_DATASET"),
+        private=True,
+    )
 else:
     print("No HF_API_TOKEN found. Logging is disabled.")
 
@@ -82,11 +85,15 @@ def translate(input_text):
     print(f"{datetime.utcnow()} | Translating: {input_text}")
 
     if getenv("HF_API_TOKEN") is not None:
-        if log_queue is None:
-            log_queue = Queue()
-            t = Thread(target=check_thread, args=(log_queue,))
-            t.start()
-        log_queue.put([input_text])
+        try:
+            logging_callback = log_data(
+                hf_token=getenv("HF_API_TOKEN"),
+                dataset_name=getenv("OUTPUT_DATASET"),
+                private=True,
+            )
+            logging_callback([input_text])
+        except:
+            print("Error happened while pushing data to HF.")
 
     input_text = f"[INST] {input_text} [/INST]"
     inputs = tokenizer([input_text], return_tensors="pt").to(model.device)