Update app.py

app.py

@@ -2,15 +2,20 @@
 import gradio as gr
 import requests
 from datetime import datetime
-from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+import os
 
-# Load Google Gemma 모델 (2B)
-model_id = "google/gemma-2b-it"
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(model_id)
+# ✅ Hugging Face token 인증 (필요시 환경변수 사용)
+from huggingface_hub import login
+login(token=os.environ.get("HF_TOKEN"), new_session=False)
+
+# ✅ Gemma 모델 로드 (gated repo)
+model_id = "google/gemma-2-2b-it"
+tokenizer = AutoTokenizer.from_pretrained(model_id, token=os.environ.get("HF_TOKEN"))
+model = AutoModelForCausalLM.from_pretrained(model_id, token=os.environ.get("HF_TOKEN"))
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-# NEIS API
+# ✅ NEIS API
 NEIS_KEY = "a69e08342c8947b4a52cd72789a5ecaf"
 SCHOOL_INFO_URL = "https://open.neis.go.kr/hub/schoolInfo"
 SCHEDULE_URL = "https://open.neis.go.kr/hub/SchoolSchedule"
@@ -104,6 +109,6 @@ with gr.Interface(
     ],
     outputs=gr.Textbox(label="GPT의 응답"),
     title="📅 학사일정 + GPT 챗봇 (Gemma 2B)",
-    description="KoAlpaca 대신 Gemma 모델로 학사일정 기반 챗봇을 실행합니다."
+    description="Gemma 모델 기반의 학사일정 챗봇입니다. 연도/월은 생략 시 오늘 기준으로 처리됩니다."
 ) as app:
     app.launch()