Update app.py

app.py

@@ -9,18 +9,25 @@ from transformers import pipeline
 
 pipe = pipeline("question-answering", model="Intel/dynamic_tinybert")
 
-# Define the function to answer questions
 def get_answer(text):
     result = pipe(question=text, context=knowledge_base_text)
-    return result["answer"]
+    answer = result["answer"]
+    confidence = result["score"]
+    return answer, confidence
 
 # Define the Gradio interface
 def chatbot_interface(Question):
-    answer = get_answer(Question)
-    return answer
+    answer, confidence = get_answer(Question)
+    if confidence < 0.2:
+        response = "Not found in the knowledge base"
+    else:
+        response = f"Answer: {answer}"
+    return response
+
 
 knowledge_base_text = """
-3D+ :  ; 
+CTC : Ceylon Tobacco Company ; 
+PTC : Pakistan Tobacco Company ; 
 3GGT : The 3 Great Growth Things ; 
 4D : Four "D" model ; 
 60 day crops : 60 day crops ; 
@@ -2185,10 +2192,6 @@ ZW06 : BAT Zimbabwe - Harare ;
 APSCSC : Asia Pacific Supply Chain Service Centre ; 
 NAA : North Asia Area ; 
 
-
-- Service desk is located at HO 1st floor & Annes near basketball ground.
-- for SAP , wifi , network related issue , you should contact Md. Monirul Islam & his contact number 01730372426
-- CTC stands for Ceylon Tobacco Company PLC.
 """
 
 iface = gr.Interface(