update app

app.py

@@ -1,9 +1,31 @@
 import gradio as gr
+from ultralyticsplus import YOLO, render_result
 
+# load model
+model = YOLO("keremberke/yolov8m-hard-hat-detection")
 
-def greet(name):
-    return "Hello " + name + "!!"
+# set model parameters
+model.overrides["conf"] = 0.25  # NMS confidence threshold
+model.overrides["iou"] = 0.45  # NMS IoU threshold
+model.overrides["agnostic_nms"] = False  # NMS class-agnostic
+model.overrides["max_det"] = 1000  # maximum number of detections per image
 
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+def get_result(img):
+    results = model.predict(img)
+    return render_result(model=model, image=img, result=results[0])
+
+
+title = "YOLO"
+description = "swag"
+examples = ["https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg"]
+
+iface = gr.Interface(
+    title=title,
+    description=description,
+    examples=examples,
+    fn=get_result,
+    inputs=gr.components.Image(shape=(512, 512)),
+    outputs=gr.components.Image(shape=(512, 512)),
+)
 iface.launch()