Upload 2 files

app.py

@@ -0,0 +1,28 @@
+# app.py
+from fastai.text.all import load_learner
+import gradio as gr
+
+# 학습된 모델 파일 로드 ('model.pkl'은 실행 디렉토리에 위치)
+learn = load_learner('model.pkl')
+
+# 한국어 질병 목록 (모델 출력 순서에 맞춰 설정)
+categories = ['알레르기', '빈혈', '기관지염', '당뇨병', '설사', '피로', '독감', '말라리아', '스트레스']
+
+def classify_korean(txt):
+    """
+    한국어로 입력된 증상을 받아 각 질병별 확률을 반환합니다.
+    """
+    _, _, probs = learn.predict(txt)
+    return {cat: float(probs[i]) for i, cat in enumerate(categories)}
+
+# Gradio 인터페이스 설정
+iface = gr.Interface(
+    fn=classify_korean,
+    inputs=gr.Textbox(lines=3, label='증상을 자세히 입력하세요'),
+    outputs=gr.Label(num_top_classes=len(categories), label='질병별 확률'),
+    examples=['목이 아프고 기침이 나요', '열이 나고 몸살 기운이 있어요'],
+    description='증상을 입력하면 해당 질병별 확률을 한국어로 보여줍니다.'
+)
+
+if __name__ == '__main__':
+    iface.launch()

requirements.txt

@@ -0,0 +1,8 @@
+# requirements.txt
+fastai>=2.7.0
+fastcore>=1.5.0
+torch>=2.0.0
+konlpy>=0.6.0
+mecab-python3>=1.0.0
+sentencepiece>=0.1.96
+gradio>=4.16.0