Create app.py

app.py

@@ -0,0 +1,22 @@
+import gradio as gr
+from ultralytics import YOLO
+from PIL import Image
+import torch
+
+# Load model
+model = YOLO("best.pt")
+
+def detect_gingivitis(image):
+    results = model(image)
+    annotated = results[0].plot()  # draw bounding boxes
+    return Image.fromarray(annotated)
+
+# Create Gradio interface
+interface = gr.Interface(
+    fn=detect_gingivitis,
+    inputs=gr.Image(type="pil"),
+    outputs="image",
+    title="Gingivitis Detection"
+)
+
+interface.launch()