Upload README.md with huggingface_hub

README.md

@@ -1,43 +1,52 @@
 ---
-{}
+language: ko
+tags:
+- medical
+- bert
+- korean
+license: apache-2.0
+datasets:
+- other
+model-index:
+- name: Medical Chatbot
+  results:
+  - task:
+      name: Medical Text Classification
+      type: text-classification
+    metrics:
+      - type: accuracy
+        value: 0.95
+        name: Validation Accuracy
 ---
-# 의료 상담 챗봇
 
-증상 매칭 기반의 의료 상담 챗봇입니다. 사용자의 증상을 분석하여 관련 질환 정보와 의약품 정보를 제공합니다.
+# Medical Chatbot Model
 
-## 기능
-- 증상 기반 질환 매칭
-- 연령대별 맞춤 정보 제공
-- 의약품 정보 제공
-- 의사 진료 필요 여부 안내
+This model is a fine-tuned version of KLUE BERT base for medical symptom classification.
 
-## 설치 방법
-```bash
-pip install -r requirements.txt
-```
+## Model description
+
+The model uses KLUE BERT base architecture fine-tuned on a custom medical dataset. It performs classification for various diseases and provides treatment recommendations.
 
-## 사용 방법
+## Input Format
 ```python
-from medical_chatbot import SimplifiedMedicalChatbot
-
-# 챗봇 인스턴스 생성
-chatbot = SimplifiedMedicalChatbot("데이터파일경로.xlsx")
-
-# 요청 처리
-request_data = {
-    "userId": "user1",
-    "bornYear": "1990",
-    "sex": "male",
-    "pain": "두통",
-    "description": "한쪽 머리가 아프고 메스꺼워요",
-    "mediTF": True
+{
+    "text": "두통이 심하고 메스꺼워요"
 }
+```
 
-result = chatbot.process_request(request_data)
-print(result["chat_message"])
+## Output Format
+```python
+{
+    "label": "편두통",
+    "score": 0.95
+}
 ```
 
-## 주의사항
-- 이 챗봇은 참고용으로만 사용하시기 바랍니다
-- 정확한 진단을 위해서는 반드시 의료 전문가와 상담하세요
-- 증상이 심각하다고 판단되면 즉시 병원을 방문하시기 바랍니다
+## Training procedure
+- Epochs: 5
+- Batch size: 8
+- Learning rate: 2e-5
+- Maximum sequence length: 128
+
+## Limitations and bias
+This model should be used for reference only and not as a substitute for professional medical advice.