| import gradio as gr | |
| def classify_image(image): | |
| outputs = model(image) | |
| return outputs # return ถึงจะแสดงใน UI ของ Gradio | |
| gr.Interface(fn=classify_image, inputs=gr.Image(type="pil"), outputs="text").launch() | |
| preds = model(image) # สมมติเป็น logits | |
| pred_label = organ_labels[preds.argmax()] # แปลงเป็นชื่ออวัยวะ | |
| return pred_label |