File size: 2,256 Bytes
ccd9809 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | import os
from dotenv import load_dotenv
from huggingface_hub import HfApi, login
def push_model_card():
# Tải biến môi trường
load_dotenv()
hf_token = os.getenv("HF_TOKEN")
if not hf_token:
print("❌ Lỗi: Không tìm thấy HF_TOKEN trong file .env")
return
# Đăng nhập
login(token=hf_token)
api = HfApi()
repo_id = "nguyendinhduybigtreetc/linhhuonglinux-office-van-ban-hanh-chinh"
repo_type = "model" # Assuming it's a model repo since it's an exported LoRA
try:
# Create repo if not exists
api.create_repo(repo_id=repo_id, repo_type=repo_type, exist_ok=True)
print(f"✅ Đã kết nối tới Repo: {repo_id}")
except Exception as e:
print(f"⚠️ Không thể tạo/xác minh repo (có thể nó đã tồn tại): {e}")
# Đường dẫn file
current_dir = os.path.dirname(os.path.abspath(__file__))
readme_path = os.path.join(current_dir, "README.md")
# Absolute path to the AI generated banner
banner_path = r"C:\Users\Yphuc\.gemini\antigravity\brain\aba926aa-59f9-4371-b585-53faa24e6a5d\linhhuong_office_banner_1777211004107.png"
# Push Banner Image
if os.path.exists(banner_path):
print("Đang đẩy ảnh Banner lên Hugging Face...")
api.upload_file(
path_or_fileobj=banner_path,
path_in_repo="banner.png",
repo_id=repo_id,
repo_type=repo_type,
commit_message="Add AI-generated banner for Model Card"
)
else:
print("⚠️ Không tìm thấy ảnh Banner.")
# Push README.md
if os.path.exists(readme_path):
print("Đang đẩy file README.md lên Hugging Face...")
api.upload_file(
path_or_fileobj=readme_path,
path_in_repo="README.md",
repo_id=repo_id,
repo_type=repo_type,
commit_message="Update Model Card (README) with SEO and Research standards"
)
print("🎉 Hoàn tất! Bạn có thể xem trang Model tại:")
print(f"👉 https://huggingface.co/{repo_id}")
else:
print("❌ Không tìm thấy file README.md.")
if __name__ == "__main__":
push_model_card()
|