Update app.py

app.py

@@ -1,6 +1,17 @@
 from fastai.vision.all import *
 import gradio as gr
 
+def get_x(row, is_test=False):
+    image_path = path_image_combined / (row['id_code'])
+    transformed_image = custom_transform(image_path)
+
+    # Check the label of the current image and apply augmentations if it belongs to the minority class
+    if not is_test and row['label'] == 1:
+        transformed_image = additional_augmentations(transformed_image)        
+    
+    return Image.fromarray(transformed_image)
+    pass
+
 learn = load_learner('your_model.pkl')
 
 categories = ('Glaucoma Present','Glaucoma Absent')