Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -41,12 +41,16 @@ def predict(img):
|
|
| 41 |
return results
|
| 42 |
|
| 43 |
|
|
|
|
|
|
|
|
|
|
| 44 |
iface = gr.Interface(
|
| 45 |
fn=predict,
|
| 46 |
inputs=gr.Image(type="pil"),
|
| 47 |
outputs=gr.Label(num_top_classes=5),
|
| 48 |
title="Image Classification with ONNX TensorRT",
|
| 49 |
description="Upload an image to classify it using the ONNX TensorRT model.",
|
|
|
|
| 50 |
)
|
| 51 |
|
| 52 |
if __name__ == "__main__":
|
|
|
|
| 41 |
return results
|
| 42 |
|
| 43 |
|
| 44 |
+
# Add an example image
|
| 45 |
+
example_image = "beignets-task-guide.png"
|
| 46 |
+
|
| 47 |
iface = gr.Interface(
|
| 48 |
fn=predict,
|
| 49 |
inputs=gr.Image(type="pil"),
|
| 50 |
outputs=gr.Label(num_top_classes=5),
|
| 51 |
title="Image Classification with ONNX TensorRT",
|
| 52 |
description="Upload an image to classify it using the ONNX TensorRT model.",
|
| 53 |
+
examples=[example_image], # Add the example image to the interface
|
| 54 |
)
|
| 55 |
|
| 56 |
if __name__ == "__main__":
|