Update app.py

app.py

@@ -1,46 +1,31 @@
 import gradio as gr
-from transformers import ViTFeatureExtractor, ViTForImageClassification
+from ultralytics import YOLO
 from PIL import Image
-import torch
-import numpy as np
 
-# *******************************************************************
-# ЕҢ ДӘЛ ЖӘНЕ ҚОҚЫС ЖІКТЕУГЕ АРНАЛҒАН МОДЕЛЬ ID-І (90%+ Accuracy)
-# *******************************************************************
-MODEL_ID = "keremberke/vit-base-patch16-224-full-empty-trash-bin" 
-CLASS_NAMES = ['Empty', 'Full'] 
+# Pretrained модельді жүктеу
+model = YOLO('yolov8n.pt')
 
-try:
-    feature_extractor = ViTFeatureExtractor.from_pretrained(MODEL_ID)
-    model = ViTForImageClassification.from_pretrained(MODEL_ID)
-    MODEL_LOADED = True
-    if model.config.id2label:
-        CLASS_NAMES = [model.config.id2label[i] for i in model.config.id2label]
+# Суретті тексеру функциясы
+def check_trash(img):
+    results = model(img)
+    boxes = results[0].boxes
+    output = []
+    for box in boxes:
+        label = box.cls.item()
+        conf = box.conf.item()
+        if conf > 0.5:
+            output.append(f"{results[0].names[int(label)]}: {conf*100:.1f}%")
+    if not output:
+        return "Қоқыс табылмады"
+    return "\n".join(output)
 
-except Exception as e:
-    print(f"ERROR: Model loading failed: {e}")
-    MODEL_LOADED = False
-
-# [ҚАЛҒАН КӨПЕЙЛІК КОД ӨЗГЕРІССІЗ ҚАЛАДЫ]
-
-def classify_trash_bin(image):
-    # ... (Алдыңғы жауаптағыдай қате өңдеу және болжау коды)
-    if not MODEL_LOADED:
-        return {"Error": 1.0, "Check Logs": 0.0}
-    # ... (Қалған код)
-    
-    try:
-        # ... (Болжам жасау)
-        return results
-
-    except Exception as e:
-        return {"Error": 1.0, "Check Logs": 0.0}
-
-iface = gr.Interface(
-    fn=classify_trash_bin,
-    inputs=gr.Image(type="numpy", label="SmartTrachAI Input"),
-    outputs=gr.Label(num_top_classes=2, label="Prediction"),
-    title="SmartTrachAI", 
-    description="Automated Trash Bin Status Detector."
+# Gradio интерфейсі
+app = gr.Interface(
+    fn=check_trash,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Textbox(),
+    title="SmartCity Waste Detection",
+    description="Қоқыс жәшігін, сыртқа төгілген қоқысты және түрін анықтайды"
 )
-iface.launch()
+
+app.launch()