Create app.py

app.py

@@ -0,0 +1,17 @@
+import gradio as gr
+from ultralytics import YOLO
+import cv2
+
+model = YOLO("https://huggingface.co/MiguelEscamilla/id-card-yolo/resolve/main/best.pt")
+
+def detect(image):
+    results = model(image)
+    return results[0].plot()
+
+demo = gr.Interface(
+    fn=detect,
+    inputs=gr.Image(type="filepath"),
+    outputs="image"
+)
+
+demo.launch()