Update app.py (#2)

- Update app.py (34faa5e0e9e33e7fd874e3c222a2eadbd6220135)


Co-authored-by: Aneesh D <[email protected]>

app.py

@@ -38,7 +38,7 @@ def predict(Image):
 
     with torch.no_grad():
         grade = torch.softmax(model(img.float()), dim=1)[0]
-    category = ["Normal", "Mild", "Moderate", "Severe", "Proliferative"]
+    category = ["0 - Normal", "1 - Mild", "2 - Moderate", "3 - Severe", "4 - Proliferative"]
     output_dict = {}
     for cat, value in zip(category, grade):
         output_dict[cat] = value.item()
@@ -55,6 +55,7 @@ demo = gr.Interface(
     inputs=image,
     outputs=label,
     #examples=["examples/0.png", "examples/1.png", "examples/2.png", "examples/3.png", "examples/4.png"]
+    examples=["0.png", "2.png", "4.png"]
     )
 
 if __name__ == "__main__":