Update app.py

app.py

@@ -0,0 +1,36 @@
+import os
+# Kütüphane yükleme hatalarını minimize etmek için
+os.environ['SYSTEM'] = 'spaces'
+
+import gradio as gr
+from ultralytics import YOLO
+import cv2
+import numpy as np
+
+# Yetki hatası önlemi
+os.environ['YOLO_CONFIG_DIR'] = '/tmp/Ultralytics'
+
+# Model yükle
+model = YOLO("best.pt")
+
+def predict_caries(image, conf_threshold):
+    if image is None:
+        return None
+    results = model.predict(source=image, conf=conf_threshold, iou=0.2, imgsz=640)
+    res_plotted = results[0].plot(labels=False, conf=False)
+    return cv2.cvtColor(res_plotted, cv2.COLOR_BGR2RGB)
+
+with gr.Blocks() as demo:
+    gr.Markdown("# 🦷 Dental Caries Detection System For Akansh Mani")
+    with gr.Row():
+        with gr.Column():
+            input_img = gr.Image(type="numpy", label="Upload X-Ray")
+            conf_slider = gr.Slider(0.01, 0.95, value=0.25, label="Confidence Threshold")
+            btn = gr.Button("🔍 Run AI Analysis", variant="primary")
+        with gr.Column():
+            output_img = gr.Image(type="numpy", label="Result")
+
+    btn.click(fn=predict_caries, inputs=[input_img, conf_slider], outputs=output_img)
+
+if __name__ == "__main__":
+    demo.launch()