Update app.py

app.py

@@ -38,7 +38,7 @@ def predict(Image):
 
     with torch.no_grad():
         grade = torch.softmax(model(img.float()), dim=1)[0]
-    category = ["0 - Normal", "1 - Mild", "2 - Moderate", "3 - Severe", "4 - Proliferative"]
+    category = ["Normal", "1 - Mild", "2 - Moderate", "3 - Severe", "4 - Proliferative"]
     output_dict = {}
     for cat, value in zip(category, grade):
         output_dict[cat] = value.item()