Update app.py

app.py

@@ -85,11 +85,11 @@ thresholds = {
     'text': 0.1,
     'Name tag':0.1,
     'License plate': 0.1,
-    'digital screens': 0.1,
+    'Digital screenss': 0.1,
     'signboard': 0.1,
     'Documents': 0.1,
-    'information board': 0.1,
-    'poster': 0.1,
+    'information boards': 0.1,
+    'posters': 0.1,
     'sign': 0.1,
     'Mail or envelope': 0.1,
     'logo': 0.1,
@@ -204,15 +204,16 @@ from datetime import datetime
 from PIL import Image
 from ultralytics import YOLO
 
-# オブジェクト選択を動的に行う関数
 def decide_to_object(risk_level):
+    # `tex` の要素を `thresholds` のキーに合わせて書き換え
     tex = [
-        'text', 'poster', 'Name tag', 'License plate', 'digital screens',
+        'text', 'posters', 'Name tag', 'License plate', 'Digital screenss',
         'signboard', 'sign', 'logo', 'manhole', 'electricity pole', 'cardboard'
     ]
-    risk_level = int(risk_level / 20) * (len(tex) / 5)  # 個数決定(1/2)
-    return tex[:int(risk_level) + 1]
-
+    
+    # リスクレベルに応じたオブジェクト数の決定
+    num_objects = int(risk_level / 20) * (len(tex) // 5)  # 個数決定(1/2)
+    return tex[:int(num_objects) + 1]
 
 # マスクを生成する関数
 def create_mask(image, x1, y1, x2, y2):