adding label func

app.py

@@ -1,7 +1,12 @@
 import gradio as gr
 from fastai.vision.all import * 
 
-def is_cat(x): return x[0].isupper()
+def label_func(fname):
+    pattern = r'(FRESH|HALF-FRESH|SPOILED)-\d+-'
+    match = re.search(pattern, fname.name)
+    if match: 
+        label = match.group(1)
+        return label
 
 learn = load_learner("model.pkl")