pipeline_config.json

pipeline_config = {
    "task": "text-generation",
    "framework": "pt",
    "model": {
        "name": "ombhojane/wellness-mini",
        "type": "AutoModelForCausalLM"
    },
    "tokenizer": {
        "name": "ombhojane/wellness-mini",
        "type": "AutoTokenizer"
    }
}

# Save pipeline config
import json
with open("pipeline_config.json", "w") as f:
    json.dump(pipeline_config, f, indent=2)

# Upload pipeline config
api.upload_file(
    path_or_fileobj="pipeline_config.json",
    path_in_repo="pipeline_config.json",
    repo_id="ombhojane/wellness-mini",
    repo_type="model"
)