| # モデルをローカルにダウンロードして保存 | |
| from transformers import AutoImageProcessor, AutoModelForImageClassification | |
| import os | |
| model_name = "krmin/mahjong_soul_vision" | |
| local_model_path = "./vision_transformer_local" | |
| print(f"モデルをダウンロード中: {model_name}") | |
| print(f"保存先: {local_model_path}") | |
| # モデルとプロセッサをダウンロード | |
| processor = AutoImageProcessor.from_pretrained(model_name) | |
| model = AutoModelForImageClassification.from_pretrained(model_name) | |
| # ローカルに保存 | |
| print("ローカルに保存中...") | |
| processor.save_pretrained(local_model_path) | |
| model.save_pretrained(local_model_path) | |
| print("✓ 完了!") | |
| print(f"\n次回からは以下のように読み込めます:") | |
| print(f'pipe = pipeline("image-classification", model="{local_model_path}", device=device)') | |