Update app.py

app.py

@@ -1,11 +1,13 @@
 from fastai.vision.all import *
 import gradio as gr
 
-# Load your models here. Example:
-# models = {'Ultrasound': load_learner('path_to_ultrasound_model.pkl'),
-#           'OCT': load_learner('path_to_oct_model.pkl'),
-#           'Fundus': load_learner('path_to_fundus_model.pkl'),
-#           'Fluorescence': load_learner('path_to_fluorescence_model.pkl')}
+# Load the models into a dictionary
+models = {
+    'Ultrasound': load_learner('ODDUltrasound.pkl'),
+    'OCT': load_learner('ODDOCT.pkl'),
+    'Fundus': load_learner('ODDfundus.pkl'),
+    'Fluorescence': load_learner('ODDfluorescence.pkl')
+}
 
 modality_keys = ['Ultrasound', 'OCT', 'Fundus', 'Fluorescence']
 
@@ -13,7 +15,7 @@ def classify_images(img_ultrasound, img_oct, img_fundus, img_fluorescence):
     imgs = [img_ultrasound, img_oct, img_fundus, img_fluorescence]
     predictions = []
     
-    # Convert images to PILImage and Predict with each model
+    # Convert images to PILImage and predict with each model
     for img, key in zip(imgs, modality_keys):
         pil_img = PILImage.create(img)
         pred, _, _ = models[key].predict(pil_img)