Update app.py

app.py

@@ -29,21 +29,13 @@ def predict(image):
     results = obj_detector(train_dataset[0]["image"])
     return plot_results(image)
 
-
-
-results = obj_detector(image)
-plot_results(image, results)
-
-def greet(name):
-    return "Hello " + name + "!!"
-
 title = "Do you have your helmet on?"
 description = """
 DETR model finetuned on "anindya64/hardhat" for hard hats detection.
 """
 
 demo = gr.Interface(
-    fn=greet, 
+    fn=predict, 
     inputs=gr.Image(type="filepath", label="Input Image"), 
     outputs="image",
     title=title,