Update README.md

README.md

@@ -101,16 +101,50 @@ datasets:
 You can initialize the model and processor for inference with `pipeline` as follows.
 
 ```python
-from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
-
-MODEL_ID = "Devocean-06/Spam_Filter-gemma"
-tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
-model = AutoModelForSequenceClassification.from_pretrained(MODEL_ID)
-pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
-
-text = "무료 쿠폰 지급! 지금 바로 클릭하세요 👉 https://spam.link 해당 문자 스팸인가요?" 
-result = pipe(text, top_k=2)
-print(result)
+from openai import OpenAI
+
+client = OpenAI(
+    base_url="model-endpoint",
+    api_key="api-key"  # vllm은 API key가 필요 없지만 필드는 필수
+)
+
+SYSTEM_PROMPT = """당신은 스팸 문자로 판정한 근거를 생성하는 대형 언어 모델입니다.
+아래 기준에 따라 스팸여부 판정의 근거를 간단명료하게 한 문장으로 작성해 주세요. 출력 포맷은 XAI 설명에 적합하도록 일관성 있게 템플릿 형식으로 고정되어야 하며, 스팸 여부 및 그 근거를 명쾌하게 제시해야 합니다.
+
+**1. 판정 근거(한 문장, 템플릿):**
+- **개인 정보 요구:** 신분증, 비밀번호, 카드 번호 등 개인 정보를 요구했기 때문입니다.
+- **기타 특이사항:** 위 항목 외에 스팸으로 의심되는 다른 패턴이 있습니다.
+- **발신자/수신자:** 발신 번호가 일반적이지 않거나 불분명하기 때문입니다.
+- **내용의 목적:** 금융 상품, 대출, 도박, 투자, 불법 복제 등의 홍보나 권유가 포함되어 있기 때문입니다.
+- **심리적 압박:** 긴급성, 공포, 호기심을 유발하여 즉각적인 행동을 유도했기 때문입니다. (예: "기간 한정", "지금 즉시", "클릭하지 않으면 불이익")
+- **링크/URL:** 일반적이지 않은 짧은 URL, 단축 URL 또는 의심스러운 링크가 포함되어 있기 때문입니다.
+
+**2. 필수 조건**
+- 반드시 출력 형식에 따라서 [스팸 판정 이유] 템플릿을 사용해야 합니다.
+- 스팸으로 판정한 이유에 대해서 구체적인 이유로 100자 이상으로 설명해야 합니다.
+- 반드시 위 판정 근거를 먼저 언급한 뒤에 출력 형식에 맞게 스팸 판정 이유를 생성해야 합니다.
+- 스팸 판정 이유 생성 시, 위 스팸 문자는 ~~ 으로 시작해야합니다.
+- 그리고 전제조건은 모두 스팸 문자로 분류된 형식이니 스팸이 아니라고 언급하면 안됩니다.
+
+### 출력 형식 예시
+- 판정 근거 : 개인정보 요구
+- 스팸 판정 이유: 위 스팸 문자는 개인정보를 요구하는 스팸으로 아파트 분양 및 부동산 투자 권유가 포함되어 있으며, 긴급성을 강조하여 즉각적인 행동을 유도하고 있습니다."""
+
+response = client.chat.completions.create(
+        model="Devocean-06/Spam_Filter-gemma",
+        messages=[
+            {"role": "system", "content": SYSTEM_PROMPT},
+            {"role": "user", "content": user_message}
+        ],
+        temperature=0.7,
+        max_tokens=2048
+    )
+print(response.choices[0].message.content)
+
+🧠 Example Output
+- 판정 근거: 내용의 목적  
+- 스팸 판정 이유: 위 스팸 문자는 금융 상품과 대출 관련 권유 내용을 포함하고 있으며,
+  ‘지금 바로’, ‘즉시 신청’과 같은 심리적 압박 어구를 사용하여 수신자의 행동을 유도하고 있습니다.
 ```
 
 ## Running with vLLM