Update app.py

app.py

@@ -140,7 +140,7 @@ def tott(model_name,url_input,image_input,threshold,type2):
     keep = processed_outputs["scores"] > threshold
     det_lab = processed_outputs["labels"][keep].tolist() 
     xxresult=0
-    if det_lab.count(1) == 0:
+    if det_lab.count(6) == 0:
         xxresult=1
     if det_lab.count(4) == 0:
         if type2=="Trench Depth Measurement":
@@ -173,7 +173,7 @@ This application can be used as follows:
 - Press Results
 """
 
-models = ["omarhkh/detr-finetuned-omar8"]
+models = ["omarhkh/CutLER-finetuned-12","omarhkh/detr-finetuned-omar8"]
 types_class = ["Trench Depth Measurement", "Trench Width Measurement"]
 
 css = '''