joheras commited on
Commit
e4394e9
1 Parent(s): fb13259

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -24,15 +24,15 @@ def classify_image(image):
24
  return {labels[i]: float(prediction[i]) for i in range(100)}
25
 
26
  image = gr.inputs.Image()
27
- label = gr.outputs.Label(num_top_classes=3)
28
 
29
  iface = gr.Interface(classify_image,image,label,
30
  #outputs=[
31
  # gr.outputs.Textbox(label="Engine issue"),
32
  # gr.outputs.Textbox(label="Engine issue score")],
33
  examples=["dog4.png","airplane4.png"],
34
- #, title="Classification of Ford Motor data",
35
- # description = "Model for predicting issues in Ford engines.",
36
  article = "Author: <a href=\"https://huggingface.co/joheras\">Jónathan Heras</a>"
37
  # examples = ["sample.csv"],
38
  )
 
24
  return {labels[i]: float(prediction[i]) for i in range(100)}
25
 
26
  image = gr.inputs.Image()
27
+ label = gr.outputs.Label(num_top_classes=5)
28
 
29
  iface = gr.Interface(classify_image,image,label,
30
  #outputs=[
31
  # gr.outputs.Textbox(label="Engine issue"),
32
  # gr.outputs.Textbox(label="Engine issue score")],
33
  examples=["dog4.png","airplane4.png"],
34
+ , title="Image classification on CIFAR-00ta",
35
+ description = "Model for predicting issues in Ford engines.",
36
  article = "Author: <a href=\"https://huggingface.co/joheras\">Jónathan Heras</a>"
37
  # examples = ["sample.csv"],
38
  )