categories

app.py

@@ -17,7 +17,8 @@ learn = load_learner('model.pkl')
 image = gr.Image()
 label = gr.Label()
 
-categories = ['Abrasions', 'Bruises', 'Burns', 'Cut', 'Diabetic Wounds', 'Laseration', 'Normal', 'Pressure Wounds', 'Surgical Wounds', 'Venous Wounds']
+categories = ['Abrasion', 'Bruise', 'Burn', 'Cut', 'Diabetic Wound', 'Laseration', 'Normal', 'Pressure Wound',
+              'Surgical Wound', 'Venous Wound']
 
 
 iface = gr.Interface(fn=classify_image, inputs=image, outputs=label)