Update app.py

app.py

@@ -3,8 +3,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
 app = FastAPI()
 
-# モデルのロード
-MODEL_NAME = "Lightblue/DeepSeek-R1-Distill-Qwen-7B-Japanese"
+# 🚀 軽量モデルに変更（7B → 3.6B）
+MODEL_NAME = "rinna/japanese-gpt-neox-3.6b"
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer)