Handle zeroGPU

app.py

@@ -47,11 +47,9 @@ if getenv("HF_API_TOKEN") is not None:
     #log_queue = Queue()
     #t = Thread(target=check_thread, args=(log_queue,))
     #t.start()
-    logging_callback = log_data(
-        hf_token=getenv("HF_API_TOKEN"),
-        dataset_name=getenv("OUTPUT_DATASET"),
-        private=True,
-    )
+
+    # handle zeroGPU
+    log_queue = None
 else:
     print("No HF_API_TOKEN found. Logging is disabled.")
 
@@ -84,15 +82,11 @@ def translate(input_text):
     print(f"{datetime.utcnow()} | Translating: {input_text}")
 
     if getenv("HF_API_TOKEN") is not None:
-        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.")
+        if log_queue is None:
+            log_queue = Queue()
+            t = Thread(target=check_thread, args=(log_queue,))
+            t.start()
+        log_queue.put([input_text])
 
     input_text = f"[INST] {input_text} [/INST]"
     inputs = tokenizer([input_text], return_tensors="pt").to(model.device)