Update app.py

app.py

@@ -98,8 +98,8 @@ def detect_objects2(model_name,url_input,image_input,threshold,type2):
 
     keep = processed_outputs["scores"] > threshold
     det_lab = processed_outputs["labels"][keep].tolist() 
-    det_lab.count(1)
-    if det_lab.count(1) > 0:
+    det_lab.count(6)
+    if det_lab.count(6) > 0:
         total_text="Trench is Detected \n Image is Not Blurry \n"
     else:
         total_text="Trench is NOT Detected \n Image is Blurry \n"