Update app.py

app.py

@@ -17,29 +17,29 @@ def start_train():
     hardware_file = open("hardware.txt", "r+").readlines()
     for hardware in hardware_file:
         hardware = hardware.strip()
-        print(f"Requested hardware is {hardware}")
+        os.system(f"echo 'Requested hardware is {hardware}'")
         curr_runtime = API.get_space_runtime(repo_id=REPO_ID)
-        print(f"Current hardware is {curr_runtime}")
+        os.system(f"echo 'Current hardware is {curr_runtime}'")
         if curr_runtime != hardware:
-            print("Trying to switch.")
+            os.system("echo 'Trying to switch.'")
             API.request_space_hardware(repo_id=REPO_ID, hardware=hardware)
         for model in model_file:
             model = model.strip()
-            print(f"Attempting to benchmark model {model}.")
+            os.system(f"echo 'Attempting to benchmark model {model}.'")
             for task in task_file:
                 task = task.strip()
-                print(f"Attempting to benchmark model {model} on task {task}.")
+                os.system(f"echo 'Attempting to benchmark model {model} on task {task}.'")
                 # Create the name of the directory for output.
                 now = time.time()
                 run_dir = f"runs/{task}/{model}/{now}"
                 os.system(f"./entrypoint.sh {REPO_ID} {model} {task} {hardware} {run_dir}")
                 # Uploads all run output to the results dataset.
-                print(f"Uploading {run_dir} to {RESULTS_DSET}")
+                os.system(f"echo 'Uploading {run_dir} to {RESULTS_DSET}'")
                 try:
                     API.create_repo(repo_id=f"{RESULTS_DSET}", repo_type="dataset",)
                     print(f"Created results dataset repository")
                 except:
-                    print(f"Using pre-existing dataset respository.")
+                    os.system("echo 'Using pre-existing dataset respository.'")
                 API.upload_folder(folder_path=run_dir, repo_id=f"{RESULTS_DSET}", repo_type="dataset",)
     print("Pausing space")
     API.pause_space(REPO_ID)