Update app.py

app.py

@@ -18,6 +18,8 @@ def start_train():
     #API.add_space_variable(repo_id=REPO_ID, key='STATUS', value='NOT_COMPUTING')
     #print("Pausing space")
     #API.pause_space(REPO_ID)
-    API.request_space_hardware(repo_id=REPO_ID, hardware="cpu-basic")
+    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']}