Update app.py

app.py

@@ -11,7 +11,7 @@ import tempfile
 import os
 
 # بارگذاری مدل آموزش‌دیده شما
-model = YOLO('/content/best.pt')  # مسیر مدل شما
+model = YOLO('/content/best.pt')  # اطمینان حاصل کنید که مسیر مدل در محیط Spaces صحیح است
 
 # تعریف نام کلاس‌ها به انگلیسی و فارسی
 class_names = {
@@ -64,16 +64,16 @@ def detect_and_draw_image(input_image):
         input_image_np = np.array(input_image)
         print("Image converted to NumPy array.")
 
-        # اجرای مدل روی تصویر با استفاده از آرایه NumPy
+        # اجرای مدل روی تصویر با استفاده از آرایه NumPy (RGB)
         results = model.predict(source=input_image_np, conf=0.3)
         print("Model prediction completed.")
 
         # دسترسی به نتایج OBB
-        if hasattr(results[0], 'boxes') and results[0].boxes is not None:
-            obb_results = results[0].boxes
+        if hasattr(results[0], 'obb') and results[0].obb is not None:
+            obb_results = results[0].obb
             print("Accessed obb_results.")
         else:
-            print("No 'boxes' attribute found in results[0].")
+            print("No 'obb' attribute found in results[0].")
             obb_results = None
 
         # بررسی وجود جعبه‌های شناسایی شده
@@ -161,8 +161,8 @@ def detect_and_draw_video(video_path):
             print(f"Model prediction completed for frame {frame_count}.")
 
             # دسترسی به نتایج OBB
-            if hasattr(results[0], 'boxes') and results[0].boxes is not None:
-                obb_results = results[0].boxes
+            if hasattr(results[0], 'obb') and results[0].obb is not None:
+                obb_results = results[0].obb
                 print("Accessed obb_results for frame.")
             else:
                 print("No 'obb' attribute found in results[0] for frame.")