fix: bug

.gitattributes

@@ -0,0 +1,31 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,5 @@
+.DS_Store
+yolov5s.pt
+__pycache__
+.venv
+flagged

README.md

@@ -0,0 +1,12 @@
+---
+title: Yolov5 Char
+emoji: 🌍
+colorFrom: yellow
+colorTo: purple
+sdk: gradio
+sdk_version: 4.8.0
+app_file: app.py
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,61 @@
+import gradio as gr
+from PIL import Image, ImageDraw
+import yolov5
+import json
+
+model = yolov5.load("./best.pt")
+
+def yolo(im):
+
+    results = model(im)  # inference
+
+    df = results.pandas().xyxy[0].to_json(orient="records")
+    res = json.loads(df)
+
+    im_with_boxes = results.render()[0]  # results.render() returns a list of images
+    
+    # Convert the numpy array back to an image
+    output_image = Image.fromarray(im_with_boxes)
+
+    draw = ImageDraw.Draw(im)
+
+    for bb in res:
+        xmin = bb['xmin']
+        ymin = bb['ymin']
+        xmax = bb['xmax']
+        ymax = bb['ymax']
+        draw.rectangle([xmin, ymin, xmax, ymax], outline="red", width=3)
+
+    return [
+        output_image,
+        res,
+        im,
+    ]
+
+
+inputs = gr.Image(type='pil', label="Original Image")
+outputs = [
+    gr.Image(type="pil", label="Output Image"), 
+    gr.JSON(label="Output JSON"),
+    gr.Image(type='pil', label="Output Image with Boxes"),
+]
+
+title = "YOLOv5 Character"
+description = "YOLOv5 Character Gradio demo for object detection. Upload an image or click an example image to use."
+article = "<p style='text-align: center'>YOLOv5 Character is an object detection model trained on the <a href=\"http://codh.rois.ac.jp/char-shape/\">日本古典籍くずし字データセット</a>.</p>"
+
+examples = [
+    ['『源氏物語』(東京大学総合図書館所蔵).jpg'],
+    ['『源氏物語』(京都大学所蔵).jpg'],
+    ['『平家物語』(国文学研究資料館提供).jpg']
+]
+demo = gr.Interface(yolo, inputs, outputs, title=title, description=description, article=article, examples=examples)
+
+demo.css = """
+.json-holder {
+    height: 300px;
+    overflow: auto;
+}
+"""
+
+demo.launch(share=False)

best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:772f81467ecfbf27a5c9e2b5b1b8b783b94c1db6d56b41b4e7c83996365afc8e
+size 691395070

requirements.txt

@@ -0,0 +1 @@
+yolov5