Update app.py

app.py

@@ -1,46 +1,42 @@
-import gradio as gr
 from transformers import pipeline
+import gradio as gr
 
-# KoAlpaca 모델 로드
+# 한국어 인스트럭션 모델 로드
 generator = pipeline(
     "text-generation",
-    model="beomi/KoAlpaca-Polyglot-1.1B",
+    model="lemon-mint/gemma-ko-2b-instruct-v0.51",
+    tokenizer="lemon-mint/gemma-ko-2b-instruct-v0.51",
     device_map="auto"
 )
 
-# 응답 함수 정의
-def chat_with_koalpaca(message, history):
-    if history is None:
-        history = []
-    # 대화 맥락을 간단히 합치기
-    context = "\n".join([f"User: {h[0]}\nAssistant: {h[1]}" for h in history])
-    prompt = f"{context}\nUser: {message}\nAssistant:"
-
-    # 모델로부터 응답 생성
-    response = generator(
-        prompt,
-        max_new_tokens=200,
+# 입시 정보 응답 함수
+def answer_question(prompt):
+    system_prompt = (
+        "너는 한국 대학 입시 전문가 AI야. "
+        "수능, 학생부 종합전형, 논술, 정시 등 모든 전형을 쉽게 설명해줘. "
+        "가능하면 학교명, 전형명, 지원조건 등을 구체적으로 알려줘.\n\n"
+    )
+    result = generator(
+        system_prompt + prompt,
+        max_new_tokens=300,
         temperature=0.7,
         top_p=0.9,
-        repetition_penalty=1.1,
         do_sample=True
-    )[0]["generated_text"]
+    )
+    return result[0]["generated_text"].replace(system_prompt, "").strip()
 
-    # 프롬프트 이후 텍스트만 추출
-    answer = response.split("Assistant:")[-1].strip()
-    history.append((message, answer))
-    return answer, history
-
-# Gradio 인터페이스
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("## 🦙 KoAlpaca Chatbot (Polyglot 1.1B)")
-    chatbot = gr.Chatbot(label="KoAlpaca")
-    msg = gr.Textbox(label="메시지를 입력하세요", placeholder="예: 안녕! 오늘 날씨 어때?")
-    clear = gr.Button("대화 초기화")
-
-    msg.submit(chat_with_koalpaca, [msg, chatbot], [msg, chatbot])
-    clear.click(lambda: None, None, chatbot, queue=False)
+# Gradio 인터페이스 설정
+app = gr.Interface(
+    fn=answer_question,
+    inputs=gr.Textbox(
+        lines=2,
+        label="입시 관련 질문 입력",
+        placeholder="예: 2025 수능 일정 알려줘 / 가천대 학생부 종합전형 설명"
+    ),
+    outputs=gr.Textbox(label="AI 답변"),
+    title="입시 전문 AI 챗봇",
+    description="한국 대학 입시 정보를 알려주는 챗봇입니다. 수시, 정시, 논술 등 모든 전형을 쉽게 설명해줍니다."
+)
 
-# 앱 실행
 if __name__ == "__main__":
-    demo.launch()
+    app.launch()