Update app.py

app.py

@@ -19,7 +19,8 @@ for dirname, _, filenames in os.walk('/kaggle/input'):
 
 
 import gradio as gr
-
+from fastai.vision.all import*
+categories = ('Healthy Cell', 'Leukemia', 'Sickle Cell', 'Thalassemia')
 def classify_image(img):
     pred,idx,probs = learn.predict(img)
     return dict(zip(categories, map(float,probs)))