Single func

app.py

@@ -11,7 +11,7 @@ HOST = "0.0.0.0"
 
 REPO_ID = "muellerzr/bert-base-cased-tpu-accelerate-experiments"
 
-def download_logs():
+def run_aim():
     filename = "aim_logs.zip"
     f = hf_hub_download(
         repo_id=REPO_ID,
@@ -20,13 +20,11 @@ def download_logs():
         cache_dir = "."
     )
     subprocess.run(f"unzip {f}".split())
-    subprocess.run(["cd", Path(f).name])
-
-def run_aim():
-    subprocess.run(["ls -a"])
+    subprocess.run(["ls -a"], cwd=Path(f).name)
     cmd = f"aim up --host {HOST} --port {PORT}".split()
-    subprocess.run(cmd)
-    
+    subprocess.run(cmd, cwd=Path(f).name)
+
+   
 html = f"""
 <iframe 
     src="http://{HOST}:{PORT}"
@@ -46,7 +44,6 @@ html = f"""
 """
 
 def main():
-    download_logs()
     run_aim()
 
 if __name__ == "__main__":