Update status_checker.py

status_checker.py

@@ -6,10 +6,7 @@ from huggingface_hub import upload_folder, HfFolder, delete_repo
 import sys
 import time
 import subprocess
-from huggingface_hub.utils import logging
 
-logging.set_verbosity_debug()
-logger = logging.get_logger(__name__)
 
 HfFolder().save_token(os.getenv("HF_TOKEN"))
 
@@ -18,7 +15,7 @@ output_dataset_id = "nateraw/asdf123"
 outputs_dir = Path('outputs')
 
 process_pid = os.getenv('USER_SCRIPT_PID', None)
-logger.debug(f"USER SCRIPT PID? - {process_pid} - type {type(process_pid)}")
+print(f"USER SCRIPT PID? - {process_pid} - type {type(process_pid)}")
 
 
 def process_is_complete():
@@ -33,22 +30,22 @@ def process_is_complete():
 output_file = Path('done.txt')
 
 def status_checker():
-    logger.debug("Waiting to find output_file to check if script is done running")
+    print("Waiting to find output_file to check if script is done running")
     while True:
-        logger.debug(f"Process is complete? {process_is_complete()}")
+        print(f"Process is complete? {process_is_complete()}")
         if output_file.exists():
-            logger.debug("Found the output file - Uploading the outputs from the script")
+            print("Found the output file - Uploading the outputs from the script")
             upload_folder(repo_id=output_dataset_id, folder_path=str(outputs_dir), path_in_repo='.', repo_type='dataset')
-            logger.debug("Finished uploading outputs from script. Done now!")
+            print("Finished uploading outputs from script. Done now!")
             return
-        logger.debug("Didn't find it...sleeping for 5 seconds.")
+        print("Didn't find it...sleeping for 5 seconds.")
         time.sleep(5)
 
 with gr.Blocks() as demo:
     gr.Markdown(Path('about.md').read_text())
 
 if not output_file.exists():
-    logger.debug("Didnt find output file on init, so starting thread to watch for it!")
+    print("Didnt find output file on init, so starting thread to watch for it!")
     thread = Thread(target=status_checker, daemon=True)
     thread.start()