mizoru commited on
Commit
42031f0
·
1 Parent(s): 44f276c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -3,7 +3,11 @@ import gradio as gr
3
 
4
  from fastai.vision.all import *
5
 
6
-
 
 
 
 
7
  learn = load_learner('xresnet50_pitch2.pkl')
8
 
9
  labels = learn.dls.vocab
 
3
 
4
  from fastai.vision.all import *
5
 
6
+ def get_x(df):
7
+ return df.path
8
+ def get_y(df):
9
+ return df.pattern
10
+
11
  learn = load_learner('xresnet50_pitch2.pkl')
12
 
13
  labels = learn.dls.vocab