Update app.py

app.py

@@ -26,8 +26,8 @@ def predict(image):
     obj_detector = pipeline(
         "object-detection", model="anindya64/detr-resnet-50-dc5-hardhat-finetuned"
     )
-    results = obj_detector(train_dataset[0]["image"])
-    return plot_results(image)
+    results = obj_detector(image)
+    return plot_results(image, results)
 
 title = "Do you have your helmet on?"
 description = """