Update app.py

app.py

@@ -6,8 +6,17 @@ pipeline = pipeline(task="image-classification", model="dqnguyen/vit-base_diabet
 
 def predict(image):
     predictions = pipeline(image)
-    return {p["label"]: p["score"] for p in predictions}
-
+    #return {p["label"]: p["score"] for p in predictions}
+    results = {}
+    for p in predictions:
+        if p["label"] == "MoHat":
+            results["Granulation tissue (Mô hạt)"] = p["score"]
+        elif p["label"] == "MoGiaMacNhiemKhuan":
+            results["Pseudomembranous tissue with a bacterial infection (Mô giả mạc nhiễm khuẩn)"] = p["score"]
+        elif p["label"] == "MoHoaiTu":
+            results["Necrotic tissue (Mô hoại tử)"] = p["score"]
+    return results
+    
 gr.Interface(
     predict,
     inputs=gr.inputs.Image(label="Upload an image (Tải một bức ảnh vét loét tiểu đường)", type="filepath"),