Update app.py

app.py

@@ -0,0 +1,25 @@
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+import torch
+import gradio as gr
+
+# 모델 이름 설정
+model_name = "medicalai/ClinicalBERT"
+
+# 토크나이저와 모델 로드
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+
+def predict(text):
+    # 입력 텍스트를 토크나이징
+    inputs = tokenizer(text, return_tensors="pt")
+    with torch.no_grad():
+        # 모델을 통해 예측 수행
+        logits = model(**inputs).logits
+    # 확률로 변환
+    probabilities = torch.softmax(logits, dim=1).numpy()[0]
+    # 결과 반환
+    return {label: prob for label, prob in zip(model.config.id2label.values(), probabilities)}
+
+# Gradio 인터페이스 생성 및 실행
+iface = gr.Interface(fn=predict, inputs="text", outputs="label")
+iface.launch()