Update app.py

app.py

@@ -1,37 +1,38 @@
 import streamlit as st
-from PIL import Image, ImageDraw
-from transformers import pipeline
+from PIL import Image
+from ultralytics import YOLO
 
-# 🚗 Модельді жүктеу
-pipe = pipeline("object-detection", model="harpreetsahota/car-dd-segmentation-yolov11")
+# 🚗 YOLO моделін жүктеу
+# Алдымен терминалда орнату керек: pip install ultralytics
+model = YOLO("harpreetsahota/car-dd-segmentation-yolov11")
 
-st.title("🚘 Машинаның зақымын анықтау (Object Detection)")
+st.title("🚘 Машинаның зақымын анықтау (YOLO Detection)")
 
 # 📂 Файлды жүктеу
 uploaded_file = st.file_uploader("Суретті жүктеңіз", type=["jpg","jpeg","png"])
 
-if uploaded_file is not None:
+if uploaded_file:
     image = Image.open(uploaded_file).convert("RGB")
     st.image(image, caption="Жүктелген сурет", use_column_width=True)
 
-    # 🔍 Анализ жасау
     with st.spinner("Анализ жасалуда..."):
-        preds = pipe(image)
+        # Модельді қолдану
+        results = model.predict(image)
 
-    # 📋 Нәтижелер
     st.subheader("📋 Нәтиже:")
-    for pred in preds:
-        label = pred["label"]
-        score = pred["score"]
-        box = pred["box"]
-        st.write(f"**{label}** (сенімділік: {score:.2f})")
-
-        # Bounding box салу
-        draw = ImageDraw.Draw(image)
-        xmin, ymin, xmax, ymax = box["xmin"], box["ymin"], box["xmax"], box["ymax"]
-        draw.rectangle(((xmin, ymin), (xmax, ymax)), outline="red", width=3)
-        draw.text((xmin, ymin - 10), f"{label} {score:.2f}", fill="red")
-
-    # 🖼️ Боялған суретті көрсету
-    st.image(image, caption="Болжам жасалған сурет", use_column_width=True)
 
+    # Нәтижелерді шығару
+    for r in results:
+        boxes = r.boxes
+        masks = r.masks
+
+        # Тек label + сенімділік шығару
+        for b in boxes:
+            cls = int(b.cls[0])           # класстың индексі
+            conf = float(b.conf[0])       # сенімділік
+            label = model.names[cls]      # класстың аты
+            st.write(f"**{label}** (сенімділік: {conf:.2f})")
+
+        # Bounding box/segmentation салынған суретті алу
+        plotted_img = r.plot()  # OpenCV форматы (numpy массив)
+        st.image(plotted_img, caption="Болжам жасалған сурет", use_column_width=True)