auto-retrain / src /models.py
sbrandeis's picture
sbrandeis HF staff
✨ Implement Auto-Retrain
55d6386 verified
import os
from pydantic import BaseModel
from typing import Literal
class Config(BaseModel):
target_namespace: str
input_dataset: str
input_model: str
autotrain_project_prefix: str
class WebhookPayloadEvent(BaseModel):
action: Literal["create", "update", "delete"]
scope: str
class WebhookPayloadRepo(BaseModel):
type: Literal["dataset", "model", "space"]
name: str
id: str
private: bool
headSha: str
class WebhookPayload(BaseModel):
event: WebhookPayloadEvent
repo: WebhookPayloadRepo
config = Config.parse_file(os.path.join(os.getcwd(), "config.json"))