Update app.py

app.py

@@ -38,7 +38,7 @@ codes = """001 - Vehicle Registration (New)
 031 - Tax Exemption Application (for special cases)
 032 - Deceased Owner’s Vehicle Ownership Transfer""".split("\n")
 
-
+undetected = "099 - Other/Undetected"
 
 
 model = SentenceTransformer('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2')
@@ -50,11 +50,14 @@ def respond(
     threshold
 ):
     global codes_emb
+    global undetected
+
+    undetected_code = undetected[:3]
 
-    if history and history[-1][-1][21:24] == "033": 
+    if history and history[-1][-1][21:24] == undetected_code: 
         list_his = ""
         for his in history[::-1]:
-            if his[-1][21:24] != "033":
+            if his[-1][21:24] != undetected_code:
                 break
             list_his = his[0] + "\n" + list_his
 
@@ -74,7 +77,8 @@ def respond(
     s_max = scores.argmax()
 
     if scores[s_max] < threshold:
-        request_code = "033 - Other/Undetected"
+        # request_code = "033 - Other/Undetected"
+        request_code = undetected
     else:
         request_code = codes[scores.argmax()]
 
@@ -114,15 +118,15 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
 with gr.Blocks() as demo:
     # Add header title and description
     gr.Markdown("# List of Request Numbers")
-    gr.Markdown("<br>".join(codes))
+    gr.Markdown("<br>".join(codes) + "<br>" + undetected)
     gr.Markdown("# Valid License Plate Number Criteria:")
-    gr.Markdown("(1-2 letter) (4 numbers) (1-3 letters)")
+    gr.Markdown("(1-2 letters) (4 numbers) (1-3 letters)")
 
     # Add chat interface
     chat_interface = gr.ChatInterface(
         respond, 
         additional_inputs=[
-            gr.Number(0.5, label="threshold for Undetected", show_label=True, minimum=0., maximum=1.0, step=0.1)
+            gr.Number(0.5, label="confidence threshold", show_label=True, minimum=0., maximum=1.0, step=0.1)
         ]
     )