Update app.py

app.py

@@ -1,25 +1,45 @@
-from transformers import AutoTokenizer, AutoModelForQuestionAnswering, pipeline
 import gradio as gr
+from transformers import pipeline
 
-model_name = "nlpai-lab/kobert-base-v1"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForQuestionAnswering.from_pretrained(model_name)
+# ✅ 한국어 경량 모델 (빠르고 공개됨)
+generator = pipeline(
+    "text-generation",
+    model="EleutherAI/polyglot-ko-1.3b",
+    device_map="auto"
+)
 
-generator = pipeline("text-generation", model="beomi/KoAlpaca-Polyglot-5.8B", device_map="auto")
+# ✅ 입시 정보 응답 함수
+def college_ai(question):
+    prompt = f"""너는 한국 입시 전문가야.
+학생이 질문하면 간결하고 정확하게 설명해줘.
+가능하면 대학명, 전형명, 반영비율, 경쟁률 같은 실제 정보를 포함해줘.
 
+질문: {question}
+답변:"""
+    response = generator(
+        prompt,
+        max_new_tokens=200,
+        do_sample=True,
+        temperature=0.7,
+        top_p=0.9,
+        repetition_penalty=1.1
+    )
+    answer = response[0]["generated_text"].split("답변:")[-1].strip()
+    return answer
 
-context = """
-수능은 매년 11월 셋째 주 목요일에 시행됩니다.
-한국장학재단은 대학생에게 국가장학금과 학자금대출을 제공합니다.
-정시모집은 수능 성적 위주로, 수시모집은 학생부나 면접으로 선발합니다.
-"""
+# ✅ Gradio 인터페이스
+demo = gr.Interface(
+    fn=college_ai,
+    inputs=gr.Textbox(
+        lines=2,
+        placeholder="예: 한국공학대 학생부교과 전형 경쟁률 알려줘",
+        label="입시 관련 질문 입력"
+    ),
+    outputs=gr.Textbox(label="AI 답변"),
+    title="🎓 입시정보 AI (College Info Assistant)",
+    description="한국 대학 입시, 학생부종합, 논술, 교과 전형 정보를 알려주는 인공지능입니다.",
+    theme="default"
+)
 
-def answer(question):
-    result = qa_pipeline(question=question, context=context)
-    return result["answer"]
-
-gr.Interface(fn=answer,
-             inputs="text",
-             outputs="text",
-             title="입시 도우미 AI",
-             description="입시 관련 질문을 해보세요. 예: '수능은 언제야?'").launch()
+if __name__ == "__main__":
+    demo.launch()