Upload 10 files

.gitattributes

@@ -32,3 +32,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+sample/1.jpg filter=lfs diff=lfs merge=lfs -text
+sample/6.jpg filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,4 @@
+venv/
+flagged/
+.idea/
+__pycache__/

app.py

@@ -0,0 +1,44 @@
+import glob
+
+import gradio as gr
+import yolov5
+from PIL import Image
+# import torch
+from huggingface_hub import hf_hub_download
+
+# Add new models here
+model_names = [
+    "TW_plate_number",
+]
+
+def load_model(model_name):
+    model_path = hf_hub_download(repo_id=model_name, filename="plate.pt")
+    return model_path
+
+def yolo_inference(im, model_path, threshold=0.70):
+    model_loaded = load_model(model_path)
+    model = yolov5.load(model_loaded)
+    model.conf = threshold
+    results = model(im)  # inference
+    numpy_image = results.render()[0]
+    output_image = Image.fromarray(numpy_image)
+    return output_image
+
+title = "TW_plate_number"
+
+inputs = [gr.Image(type="pil", label="document image"),
+          gr.inputs.Dropdown(model_names, label="Model", default=model_names[0]),
+          gr.Slider(maximum=1, step=0.01, value=0.70)]
+
+
+examples = [[str(file),model_names[0], 0.70] for file in glob.glob("./sample/*.jpg")]
+
+demo=gr.Interface(fn=yolo_inference,
+                  inputs=inputs,
+                  outputs=gr.Image(type="pil", label="annotated document").style(height=800),
+                  title=title,
+                  theme="huggingface",
+                  examples=examples)
+
+if __name__ == "__main__":
+    demo.launch(debug=True)

gitattributes.txt

@@ -0,0 +1,35 @@
+*.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
+*.ckpt 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
+*.mlmodel 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
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl 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
+*.safetensors 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
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text

requirements.txt

@@ -0,0 +1,2 @@
+yolov5==7.0.2
+gradio