Spaces:
Runtime error
Runtime error
imkaushalpatel
commited on
Commit
•
0d4af41
1
Parent(s):
c09d2e5
Update app.py
Browse files
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="
|
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."
|