Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| from transformers import DetrImageProcessor, DetrForObjectDetection | |
| processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50") | |
| model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50") | |
| def detect_objects(image): | |
| inputs = processor(images=image, return_tensors="pt") | |
| outputs = model(**inputs) | |
| boxes = outputs.pred_boxes | |
| labels = outputs.pred_classes | |
| detected_objects = [f"{label}: {box}" for label, box in zip(labels, boxes)] | |
| return detected_objects | |
| inputs = gr.inputs.Image(label="Upload Image", type="pil") | |
| outputs = gr.outputs.Textbox(label="Detected Objects") | |
| gr.Interface(detect_objects, inputs, outputs, capture_session=True).launch() | |