[email protected] commited on
Commit
5ff0020
1 Parent(s): 2d26efc

Add title and description. Enable interpretation

Browse files
Files changed (2) hide show
  1. app.py +12 -1
  2. requirements.txt +2 -1
app.py CHANGED
@@ -26,5 +26,16 @@ label = gr.Label()
26
  # Upload your own images and link them
27
  examples = ['cardigan.jpg', 'pembroke.jpg']
28
 
29
- intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
 
 
 
 
 
 
 
 
 
 
 
30
  intf.launch()
 
26
  # Upload your own images and link them
27
  examples = ['cardigan.jpg', 'pembroke.jpg']
28
 
29
+ title = "Corgi Breed Classifier"
30
+ description = "A Corgie breed classifier to distinguish between Welsh Corgi Pembroke and Welsh Corgi Cardigan."
31
+ interpretation='default'
32
+
33
+ intf = gr.Interface(
34
+ fn=classify_image,
35
+ inputs=image,
36
+ outputs=label,
37
+ examples=examples,
38
+ title=title,
39
+ description=description,
40
+ interpretation=interpretation)
41
  intf.launch()
requirements.txt CHANGED
@@ -2,4 +2,5 @@ fastai
2
  gradio
3
  timm
4
  torch
5
- torchvision
 
 
2
  gradio
3
  timm
4
  torch
5
+ torchvision
6
+ scikit-image