Update app.py

app.py

@@ -38,7 +38,7 @@ def predict_skin_disease(image):
         _, predicted_class = torch.max(output, 1)
 
     # Map the predicted class index to the corresponding class label
-    class_label =  ['AbdomenCT', 'BreastMRI', 'CXR', 'ChestCT', 'Hand', 'HeadCT']
+    class_label =  ['No DR', 'Mild', 'Moderate', 'Severe', 'Proliferative']
     class_label = class_label[predicted_class.item()]