Zakia commited on
Commit
1c16f01
1 Parent(s): e8e41b9

add cat and dog images and change app.py

Browse files
Files changed (7) hide show
  1. app.py +12 -5
  2. cat1.jpg +0 -0
  3. cat2.jpg +0 -0
  4. cat3.jpg +0 -0
  5. dog1.jpg +0 -0
  6. dog2.jpg +0 -0
  7. dog3.jpg +0 -0
app.py CHANGED
@@ -23,17 +23,24 @@ def predict(img):
23
  return dict(zip(categories, map(float, probs)))
24
 
25
 
26
- title = "CAT PREDICTOR"
27
 
28
- description = "A cat predictor model trained on the Pets dataset with fastai. Created as a demo for Gradio and Hugging Face spaces."
29
 
30
 
31
- article = "<p style='text-align: center'><span style='font-size: 15pt;'>CAT_PREDICTOR. 2022. </span></p>"
32
 
33
 
34
- image = gr.inputs.Image(shape=(192, 192))
35
  label = gr.outputs.Label()
36
- examples = ['chapter1_cat_example.jpg']
 
 
 
 
 
 
 
37
  interpretation = 'default'
38
  enable_queue = True
39
 
 
23
  return dict(zip(categories, map(float, probs)))
24
 
25
 
26
+ title = "Cat or Dog Predictor"
27
 
28
+ description = "A cat or dog predictor model trained on the Pets dataset with fastai."
29
 
30
 
31
+ article = "<p style='text-align: center'><span style='font-size: 15pt;'>Cat or Dog Predictor. Zakia Salod. 2022. </span></p>"
32
 
33
 
34
+ image = gr.inputs.Image(shape=(512, 512))
35
  label = gr.outputs.Label()
36
+ examples = [
37
+ ['cat1.jpg'],
38
+ ['dog1.jpg'],
39
+ ['cat2.jpg'],
40
+ ['dog2.jpg'],
41
+ ['cat3.jpg'],
42
+ ['dog3.jpg'],
43
+ ]
44
  interpretation = 'default'
45
  enable_queue = True
46
 
cat1.jpg ADDED
cat2.jpg ADDED
cat3.jpg ADDED
dog1.jpg ADDED
dog2.jpg ADDED
dog3.jpg ADDED