Update app.py

app.py

@@ -2,6 +2,20 @@ import gradio as gr
 from fastai.vision.all import *
 import skimage
 
+train_path = Path("/kaggle/input/aptos2019/train_images")
+test_path = Path("/kaggle/input/aptos2019/test_images")
+train_df = pd.read_csv("/kaggle/input/aptos2019/train_1.csv")
+
+def get_x(r):
+    filename = f"{r['id_code']}.png"
+    file_path = train_path / filename
+    if os.path.exists(file_path):
+        return str(file_path)
+    else:
+        return "/kaggle/input/aptos2019/train_images/train_images/1ae8c165fd53.png"
+        
+def get_y(r): return r['diagnosis']
+
 learn = load_learner('model.pkl')
 labels = learn.dls.vocab