Update app.py

app.py

@@ -10,16 +10,25 @@ app = FastAPI()
 
 @app.get("/")
 def start_train():
+    model_file = open("models.txt", "r+").readlines()
+    task_file = open("tasks.txt", "r+").readlines()
+    hardware_file = open("hardware.txt", "r+").readlines()
+    for hardware in hardware_file:
+        hardware = hardware.strip()
+        curr_runtime = API.get_space_runtime(repo_id=REPO_ID)
+        if curr_runtime != hardware:
+            API.request_space_hardware(repo_id=REPO_ID, hardware=hardware)
+        for model in model_file:
+            model = model.strip()
+            for task in task_file:
+                task = task.strip()
+                os.system(f"./entrypoint.sh {REPO_ID} {model} {task} {hardware}")
     #space_variables = API.get_space_variables(repo_id=REPO_ID)
     #if 'STATUS' not in space_variables or space_variables['STATUS'] != 'COMPUTING':
     #    print("Beginning processing.")
     #    API.add_space_variable(repo_id=REPO_ID, key='STATUS', value='COMPUTING')
-    os.system(f"./entrypoint.sh {REPO_ID}")
+    #os.system(f"./entrypoint.sh {REPO_ID} {model} {task} {hardware}")
     #API.add_space_variable(repo_id=REPO_ID, key='STATUS', value='NOT_COMPUTING')
-    #print("Pausing space")
-    #API.pause_space(REPO_ID)
-    curr_runtime = API.get_space_runtime(repo_id=REPO_ID)
-    if curr_runtime != "cpu-basic":
-        API.request_space_hardware(repo_id=REPO_ID, hardware="cpu-basic")
-
-    return {"Status": "whatever"}#space_variables['STATUS']}
+    print("Pausing space")
+    API.pause_space(REPO_ID)
+    return {"Status": "Done"}#space_variables['STATUS']}