Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ from transformers import AutoImageProcessor, AutoModelForObjectDetection
|
|
| 7 |
def model():
|
| 8 |
processor = AutoImageProcessor.from_pretrained("facebook/detr-resnet-50")
|
| 9 |
model = AutoModelForObjectDetection.from_pretrained("facebook/detr-resnet-50")
|
| 10 |
-
pipe = pipeline("object-detection", model=
|
| 11 |
output=pipe("dogs1.webp")
|
| 12 |
return output
|
| 13 |
|
|
|
|
| 7 |
def model():
|
| 8 |
processor = AutoImageProcessor.from_pretrained("facebook/detr-resnet-50")
|
| 9 |
model = AutoModelForObjectDetection.from_pretrained("facebook/detr-resnet-50")
|
| 10 |
+
pipe = pipeline("object-detection", model=model)
|
| 11 |
output=pipe("dogs1.webp")
|
| 12 |
return output
|
| 13 |
|