Cecilia Casarini commited on
Commit
19cd6ec
·
verified ·
1 Parent(s): 2a3915a
Files changed (4) hide show
  1. .gitattributes +1 -0
  2. orchid.jpg +0 -0
  3. vanilla.jpg +3 -0
  4. vanilla.py +4 -4
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ vanilla.jpg filter=lfs diff=lfs merge=lfs -text
orchid.jpg ADDED
vanilla.jpg ADDED

Git LFS Details

  • SHA256: d84bcfe3f5998f67570a10beec2cf2ca41b68b072746cf938247c2ac0054ac58
  • Pointer size: 132 Bytes
  • Size of remote file: 4.01 MB
vanilla.py CHANGED
@@ -2,8 +2,7 @@ from fastai.vision.all import *
2
  import gradio as gr
3
 
4
 
5
- def is_vanilla(x): return x[0].isupper()
6
- learn = load_learner(vanilla.pkl)
7
 
8
  categories = ('vanilla', 'orchid')
9
 
@@ -11,8 +10,9 @@ def classify_image(img):
11
  pred, idx, probs = learn.predict(img)
12
  return dict(zip(categories, map(float, probs)))
13
 
14
- image = gr.inputs.Image(shape = (192,192))
15
- label = gr.outputs.Label()
 
16
 
17
  examples = ['vanilla.jpg', 'orchid.jpg']
18
 
 
2
  import gradio as gr
3
 
4
 
5
+ learn = load_learner('vanilla.pkl')
 
6
 
7
  categories = ('vanilla', 'orchid')
8
 
 
10
  pred, idx, probs = learn.predict(img)
11
  return dict(zip(categories, map(float, probs)))
12
 
13
+ image = gr.Image()
14
+ label = gr.Label()
15
+
16
 
17
  examples = ['vanilla.jpg', 'orchid.jpg']
18