write_model_card.py

import json

# Nội dung của model card ở định dạng JSON
model_card = {
  "model_name": "ViSD-TDNM",
  "model_description": "Model tạo ảnh dựa trên ổn định Diffusion với checkpoint phiên bản 1.5.",
  "task": "Tạo ảnh từ văn bản",
  "data": "Tập dữ liệu đào tạo bao gồm các cặp văn bản-hình ảnh.",
  "metrics": {"mAP": 0.85, "Precision": 0.92},
  "checkpoint": {
    "path": "path/to/your/checkpoint/file.ckpt",
    "version": "1.5"
  },
  "tags": ["tạo ảnh", "ổn định Diffusion", "model tạo ảnh"]
}

# Đường dẫn đến file JSON để lưu
model_card_path = "model_card.json"

# Lưu model card vào file JSON
with open(model_card_path, "w") as file:
    json.dump(model_card, file, indent=4)

print(f"Model card đã được lưu vào {model_card_path}")