subh71 commited on
Commit
1f4d237
·
verified ·
1 Parent(s): 76a2e00

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=processor)
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