File size: 701 Bytes
a15535e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
"""Dump build + runtime logs from a Hugging Face Space for debugging."""
from __future__ import annotations

import os
import sys

import requests

TOKEN = os.environ.get("HF_TOKEN") or os.environ.get("HUGGINGFACE_TOKEN")
if not TOKEN:
    sys.exit("set HF_TOKEN")

REPO = sys.argv[1] if len(sys.argv) > 1 else "akhiilll/forgeenv-demo"
LOG_TYPE = sys.argv[2] if len(sys.argv) > 2 else "run"  # "build" | "run"

url = f"https://api.hf.space/v1/{REPO}/logs/{LOG_TYPE}"
print(f"GET {url}")
r = requests.get(url, headers={"Authorization": f"Bearer {TOKEN}"}, stream=True, timeout=30)
print(f"status={r.status_code}\n---")
for chunk in r.iter_lines(decode_unicode=True):
    if chunk:
        print(chunk)