Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -1,3 +1,21 @@
|
|
1 |
import gradio as gr
|
|
|
2 |
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
import gradio as gr
|
2 |
+
from transformers import DetrImageProcessor, DetrForObjectDetection
|
3 |
|
4 |
+
processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
|
5 |
+
model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
|
6 |
+
|
7 |
+
def detect_objects(image):
|
8 |
+
inputs = processor(images=image, return_tensors="pt")
|
9 |
+
|
10 |
+
outputs = model(**inputs)
|
11 |
+
|
12 |
+
boxes = outputs.pred_boxes
|
13 |
+
labels = outputs.pred_classes
|
14 |
+
|
15 |
+
detected_objects = [f"{label}: {box}" for label, box in zip(labels, boxes)]
|
16 |
+
return detected_objects
|
17 |
+
|
18 |
+
inputs = gr.inputs.Image(label="Upload Image", type="pil")
|
19 |
+
outputs = gr.outputs.Textbox(label="Detected Objects")
|
20 |
+
|
21 |
+
gr.Interface(detect_objects, inputs, outputs, capture_session=True).launch()
|