Antoine101 commited on
Commit
918e570
·
verified ·
1 Parent(s): 42b4f1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def plot_results(image, results, threshold=0.7):
25
  def predict(image):
26
  # make the object detection pipeline
27
  obj_detector = pipeline(
28
- "object-detection", model="anindya64/detr-resnet-50-dc5-hardhat-finetuned"
29
  )
30
  results = obj_detector(image)
31
  return plot_results(image, results)
 
25
  def predict(image):
26
  # make the object detection pipeline
27
  obj_detector = pipeline(
28
+ "object-detection", model="Antoine101/detr-resnet-50-dc5-hardhat-finetuned"
29
  )
30
  results = obj_detector(image)
31
  return plot_results(image, results)