Update app.py

app.py

@@ -83,13 +83,13 @@ dangerarray=[10,30,90,50,80,20,40,70,100,60]#ここに各クラスターの危
 
 thresholds = {
     'text': 0.1,
-    'Name tag':0.1,
+    'Name tag': 0.1,
     'License plate': 0.1,
-    'Digital screenss': 0.1,
+    'Digital screens': 0.1,
     'signboard': 0.1,
-    'Documents': 0.1,
-    'information boards': 0.1,
-    'posters': 0.1,
+    'documents': 0.1,
+    'information board': 0.1,
+    'poster': 0.1,
     'sign': 0.1,
     'Mail or envelope': 0.1,
     'logo': 0.1,
@@ -207,7 +207,7 @@ from ultralytics import YOLO
 def decide_to_object(risk_level):
     # `tex` の要素を `thresholds` のキーに合わせて書き換え
     tex = [
-        'text', 'posters', 'Name tag', 'License plate', 'Digital screenss',
+        'text', 'poster', 'Name tag', 'License plate', 'Digital screens',
         'signboard', 'sign', 'logo', 'manhole', 'electricity pole', 'cardboard'
     ]
     
@@ -215,6 +215,7 @@ def decide_to_object(risk_level):
     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):
     mask = np.zeros((image.shape[0], image.shape[1]), dtype=np.uint8)