gemma-2 / app /models.py
ZSCGR's picture
Upload folder using huggingface_hub
30c9413 verified
raw
history blame contribute delete
651 Bytes
from typing import Any, Dict, Optional
# from huggingface_hub import SpaceStage
from pydantic import BaseModel, Field
class DeployRequest(BaseModel):
hf_token: str = Field(..., description="HF token with write permission")
git_repo_url: str = Field(...)
deploy_path: str = Field("/")
space_name: str = Field(...)
space_port: int = Field(7860)
description: str = Field("")
env_vars: Dict[str, str] = Field(default_factory=dict)
private: bool = Field(False)
class DeployStatus(BaseModel):
task_id: str
status: str # PENDING | IN_PROGRESS | SUCCESS | FAILED
detail: Optional[Any] = None