chart-analyzer / model_uploader.py
tmmdev's picture
Upload 13 files
e42e5e9 verified
raw
history blame
597 Bytes
from huggingface_hub import HfApi, login
import ollama
def export_and_upload_models():
api = HfApi()
login()
# Export Ollama models
models = {
"llama3.2-vision": "tmmdev/chart-vision-analyzer",
"codellama": "tmmdev/chart-generator"
}
for model_name, repo_id in models.items():
# Export model from Ollama
model_path = ollama.export(model_name)
# Upload to HF
api.upload_folder(
folder_path=model_path,
repo_id=repo_id,
repo_type="model"
)