Update app.py

app.py

@@ -140,22 +140,19 @@ 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:
-    else:
+    if det_lab.count(1) == 0:
         xxresult=1
-    if det_lab.count(4) > 0:
-    else:
+    if det_lab.count(4) == 0:
         if type2=="Trench Depth Measurement":
             xxresult=1
-    if det_lab.count(5) > 0:
-    else:
+    if det_lab.count(5) == 0:
         if type2=="Trench Width Measurement":
             xxresult=1            
     
     if xxresult==0:
-        return "The photo is ACCEPTED"
+        return """<h1 id="title">The photo is ACCEPTED</h1>"""
     else:
-        return "The photo is NOT ACCEPTED"
+        return """<h1 id="title">The photo is NOT ACCEPTED</h1>"""
 
     
 def set_example_image(example: list) -> dict: