Ethium commited on
Commit
4ff3206
1 Parent(s): 374cd0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -1,6 +1,16 @@
1
  from fastai.vision.all import *
2
  import gradio as gr
3
 
 
 
 
 
 
 
 
 
 
 
4
  # Load the models into a dictionary
5
  models = {
6
  'Ultrasound': load_learner('ODDUltrasound.pkl'),
 
1
  from fastai.vision.all import *
2
  import gradio as gr
3
 
4
+ from pathlib import Path
5
+ import pandas as pd
6
+
7
+ def get_x(row):
8
+ # All files are assumed to be '.jpg', so we directly return the path with '.jpg' extension
9
+ return path_image_combined / f"{row['file_name']}.jpg"
10
+
11
+ def get_y(row):
12
+ return row['Buried ODD']
13
+
14
  # Load the models into a dictionary
15
  models = {
16
  'Ultrasound': load_learner('ODDUltrasound.pkl'),