Update app.py

app.py

@@ -8,21 +8,17 @@ import gradio as gr
 from fastai.vision.all import *
 from fastai.data.all import *
 
+
+def label_func(item):
+    rel_path = str(item.relative_to('dataset/train'))
+    return train_csv[train_csv['image_ID']==rel_path]["label"].values[0]
+
 # %% app.ipynb 2
 learn = load_learner('model.pkl')
 
 # %% app.ipynb 3
 categories = ('Badminton', 'Cricket', 'Karate', 'Soccer', 'Swimming', 'Tennis', 'Wrestling')
 
-# %% app.ipynb 4
-import pandas as pd
-train_csv = pd.read_csv('dataset/train.csv')
-n_inp = len(set(train_csv['label']))
-train_csv.head()
-def label_func(item):
-    rel_path = str(item.relative_to('dataset/train'))
-    return train_csv[train_csv['image_ID']==rel_path]["label"].values[0]
-
 # %% app.ipynb 5
 def classify_image(img):
     pred, idx, probs = learn.predict(img)