| |
|
|
| import subprocess |
| import os |
| from huggingface_hub import hf_hub_download |
| import gradio as gr |
| import time |
|
|
| PORT = 7860 |
| HOST = "0.0.0.0" |
|
|
| REPO_ID = "muellerzr/bert-base-cased-tpu-accelerate-experiments" |
|
|
| def download_logs(): |
| filename = "aim_logs.zip" |
| f = hf_hub_download( |
| repo_id=REPO_ID, |
| repo_type="model", |
| filename=filename, |
| cache_dir = "." |
| ) |
| subprocess.run(f"unzip {f}".split()) |
|
|
| def run_aim(): |
| cmd = f"aim up --host {HOST} --port {PORT}".split() |
| subprocess.Popen(cmd) |
|
|
| def run(): |
| print("Starting server....") |
|
|
| def main(): |
| download_logs() |
| run_aim() |
| time.sleep(5) |
|
|
| if __name__ == "__main__": |
| main() |
|
|
|
|
