daspartho commited on
Commit
660b7d4
·
1 Parent(s): 755b304

added title and description

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -9,9 +9,13 @@ def classify_image(img):
9
  return dict(zip(categories, map(float, probs)))
10
 
11
  iface = gr.Interface(
 
 
12
  fn=classify_image,
13
  inputs=gr.inputs.Image(shape=(224,224)),
14
  outputs=gr.outputs.Label(),
15
- examples=['lego.jpg', 'dog.jpg', 'cactus.jpg', 'plushie.jpg']
 
 
16
  )
17
  iface.launch()
 
9
  return dict(zip(categories, map(float, probs)))
10
 
11
  iface = gr.Interface(
12
+ title = "Is it Huggable?",
13
+ description = "An image classifier to classify things as huggable or not trained on examples of both categories.",
14
  fn=classify_image,
15
  inputs=gr.inputs.Image(shape=(224,224)),
16
  outputs=gr.outputs.Label(),
17
+ examples=['lego.jpg', 'dog.jpg', 'cactus.jpg', 'plushie.jpg'],
18
+ live=True,
19
+ enable_queue=True
20
  )
21
  iface.launch()