Upload app.py and model weights

app.py

@@ -0,0 +1,40 @@
+from ultralytics import YOLO
+import gradio as gr
+from PIL import Image, ImageDraw, ImageFont
+
+# Load a model
+model = YOLO("best.pt")
+
+
+def inference(gr_input):
+    """
+    Inference function for gradio.
+    """
+    pred = model(gr_input)
+    draw_prediction = ImageDraw.Draw(gr_input)
+    boxes_predict = pred[0].boxes
+    boxes = boxes_predict.xyxy.tolist()
+    scores = boxes_predict.conf.tolist()
+    for score, box in zip(scores, boxes):
+        x, y, x2, y2 = tuple(box)
+        draw_prediction.rectangle((x, y, x2, y2), outline="red", width=2)
+    return gr_input
+
+
+def main():
+    imagein = gr.inputs.Image(label="Input Image", type="pil")
+    imageout = gr.outputs.Image(label="Predicted Image", type="pil")
+
+    interface = gr.Interface(
+        fn=inference,
+        inputs=imagein,
+        outputs=imageout,
+        title="Potholes detection",
+        interpretation="default",
+    ).launch(debug="True", share="True")
+
+    # launch demo
+    interface.launch()
+
+if __name__ == "__main__":
+    main() 

best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3885048e6d5a451e3a6b8111b76b907c37cde4c554acbbc9104587a979692f2a
+size 6228270