Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from PIL import Image | |
| from generic import evaluate | |
| def generate_viz(image): | |
| print(f"Image: {image.size}") | |
| viz= evaluate(image, 'cpu') | |
| viz.savefig("visualization.png") | |
| return Image.open("visualization.png").convert("RGB") | |
| title = "Explain DETR 😟 for Object Detection" | |
| iface = gr.Interface(fn=generate_viz, inputs=[ | |
| gr.Image(type="pil", label="Input Image"), | |
| ], | |
| outputs=[gr.Image(label="XAI-Image", width=700, height=300)], | |
| title=title, | |
| allow_flagging="never", | |
| cache_examples=True, | |
| examples=[ | |
| ["DETR/samples/cat-remote.jpg"], | |
| ["DETR/samples/ski.jpg"], | |
| ["DETR/samples/bus.jpg"], | |
| ["DETR/samples/piza.jpg"], | |
| ], | |
| ) | |
| iface.launch(debug=True) |