Update app.py

app.py

@@ -4,13 +4,13 @@ import requests
 from datetime import datetime
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 
-# ✅ 1. KoAlpaca 모델 로드
-model_id = "beomi/KoAlpaca-Polyglot-5.8B"
+# Load Google Gemma 모델 (2B)
+model_id = "google/gemma-2b-it"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id)
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-# ✅ 2. 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"
@@ -23,7 +23,6 @@ REGIONS = {
 MONTH_NAMES = ["01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12"]
 
 
-# ✅ 3. 학교 코드 조회
 def get_school_code(region_code, school_name):
     params = {
         "KEY": NEIS_KEY,
@@ -41,7 +40,6 @@ def get_school_code(region_code, school_name):
         return None, None
 
 
-# ✅ 4. 학사일정 조회
 def get_schedule(region_code, school_code, year, month):
     from_ymd = f"{year}{month}01"
     to_ymd = f"{year}{month}31"
@@ -58,15 +56,12 @@ def get_schedule(region_code, school_code, year, month):
     res = requests.get(SCHEDULE_URL, params=params)
     data = res.json()
     try:
-        rows = data["SchoolSchedule"][1]["row"]
-        return rows
+        return data["SchoolSchedule"][1]["row"]
     except:
         return []
 
 
-# ✅ 5. GPT 응답 생성
 def generate_answer(region, school_name, year, month, question):
-    # 기본값 보정
     now = datetime.now()
     if not year:
         year = str(now.year)
@@ -98,18 +93,17 @@ def generate_answer(region, school_name, year, month, question):
     return result
 
 
-# ✅ 6. Gradio 인터페이스 정의
 with gr.Interface(
     fn=generate_answer,
     inputs=[
         gr.Dropdown(choices=list(REGIONS.keys()), label="교육청 선택"),
         gr.Textbox(label="학교명 입력"),
-        gr.Textbox(label="년도 입력 (선택, 예: 2025)"),
-        gr.Dropdown(choices=MONTH_NAMES, label="월 선택 (선택, 예: 07)"),
+        gr.Textbox(label="년도 입력 (예: 2025, 비워두면 현재 연도 사용)"),
+        gr.Dropdown(choices=MONTH_NAMES, label="월 선택 (예: 07, 비워두면 현재 월 사용)"),
         gr.Textbox(label="질문 입력 (예: 여름방학은 언제야?)")
     ],
     outputs=gr.Textbox(label="GPT의 응답"),
-    title="📅 학사일정 + GPT 챗봇 (KoAlpaca)",
-    description="학교명과 질문을 입력하면 GPT가 학사일정 기반으로 대답해줍니다. 연도/월은 비워두면 오늘 기준으로 처리됩니다."
+    title="📅 학사일정 + GPT 챗봇 (Gemma 2B)",
+    description="KoAlpaca 대신 Gemma 모델로 학사일정 기반 챗봇을 실행합니다."
 ) as app:
     app.launch()