ml_api/apps.py

# class MlApiConfig(AppConfig):
#     default_auto_field = "django.db.models.BigAutoField"
#     name = "ml_api"
#     tokenizer = None
#     model = None
#     def ready(self):
#         if MlApiConfig.model is None:
#             model_name = "rinna/japanese-gpt-neox-small"
#             print("Loading Model... (Download starts on first run)")
#             # トークナイザーとモデルのロード
#             MlApiConfig.tokenizer = AutoTokenizer.from_pretrained(
#                 model_name, use_fast=False
#             )
#             MlApiConfig.model = AutoModelForCausalLM.from_pretrained(model_name)
#             print("Model Loaded!")
from django.apps import AppConfig


class MlApiConfig(AppConfig):
    default_auto_field = "django.db.models.BigAutoField"
    name = "ml_api"