Update app.py

app.py

@@ -6,12 +6,12 @@ def is_cat(x):returnx[0].isupper()
 
 learn=load_learner('HAMM1.pkl')
 
-categories=('akiec','bcc','bkl','df','mel','nv','vasc')
+categories=learn.dls.vocab
 
 def classify_img(img):
     pred,idx,probs=learn.predict(img)
     return dict(zip(categories,map(float,probs)))
-    
+  
 image =gr.inputs.Image(shape=(192,192))
 label =gr.outputs.Label()
 example=['mel.jpg','akiec.jpg','vasc.jpg']