imkaushalpatel commited on
Commit
0d4af41
1 Parent(s): c09d2e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -41,8 +41,8 @@ def inference(input_image):
41
  result[categories[top5_catid[i]]] = top5_prob[i].item()
42
  return result
43
 
44
- inputs = gr.inputs.Image(type='pil')
45
- outputs = gr.outputs.Image(type="confidences",num_top_classes=5)
46
 
47
  title = "GOOGLENET"
48
  description = "Gradio demo for GOOGLENET, GoogLeNet was based on a deep convolutional neural network architecture codenamed Inception which won ImageNet 2014. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below."
 
41
  result[categories[top5_catid[i]]] = top5_prob[i].item()
42
  return result
43
 
44
+ inputs = gr.inputs.Image(type='pil', label="Original Image")
45
+ outputs = gr.outputs.Image(type="pil", label="Output Image", num_top_classes=5)
46
 
47
  title = "GOOGLENET"
48
  description = "Gradio demo for GOOGLENET, GoogLeNet was based on a deep convolutional neural network architecture codenamed Inception which won ImageNet 2014. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below."