Update app.py

Change disease name

app.py

@@ -10,13 +10,23 @@ import gradio as gr
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # ==============================
-# 類別名稱（14 類，對應你的 best_model.pth）
+# 類別名稱（14 類英文疾病）
 # ==============================
 class_names = [
-    "disease_1", "disease_2", "disease_3", "disease_4",
-    "disease_5", "disease_6", "disease_7", "disease_8",
-    "disease_9", "disease_10", "disease_11", "disease_12",
-    "disease_13", "disease_14"
+    "Actinic keratoses",
+    "Basal cell carcinoma",
+    "Benign keratosis-like-lesions",
+    "Chickenpox",
+    "Cowpox",
+    "Dermatofibroma",
+    "Healthy",
+    "HFMD",
+    "Measles",
+    "Melanocytic nevi",
+    "Melanoma",
+    "Monkeypox",
+    "Squamous cell carcinoma",
+    "Vascular lesions"
 ]
 num_classes = len(class_names)  # 14
 
@@ -59,8 +69,8 @@ iface = gr.Interface(
     fn=predict,
     inputs=gr.Image(type="pil"),
     outputs=gr.Label(num_top_classes=1),
-    title="疾病分類器",
-    description="上傳一張圖片，模型會判斷這是什麼疾病（14 類疾病）。"
+    title="Skin Disease Classifier",
+    description="Upload an image, and the model will predict the skin disease (14 categories)."
 )
 
 iface.launch()