|
|
|
|
|
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() |
|
|
|
|
