Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ def predict(Image):
38
 
39
  with torch.no_grad():
40
  grade = torch.softmax(model(img.float()), dim=1)[0]
41
- category = ["0 - Normal", "1 - Mild", "2 - Moderate", "3 - Severe", "4 - Proliferative"]
42
  output_dict = {}
43
  for cat, value in zip(category, grade):
44
  output_dict[cat] = value.item()
 
38
 
39
  with torch.no_grad():
40
  grade = torch.softmax(model(img.float()), dim=1)[0]
41
+ category = ["Normal", "1 - Mild", "2 - Moderate", "3 - Severe", "4 - Proliferative"]
42
  output_dict = {}
43
  for cat, value in zip(category, grade):
44
  output_dict[cat] = value.item()