Update agent_t5.py

agent_t5.py

@@ -9,6 +9,7 @@ from langchain.document_loaders import PyPDFLoader
 
 from transformers import AutoTokenizer, T5ForConditionalGeneration
 from retrieval.retrieval import Retrieval, BM25
+from datetime import datetime
 import os, time, torch
 from torch.nn import Softmax
 import requests
@@ -57,6 +58,10 @@ class Agent:
             return self.API_call(prompt)
 
     def asking(self, question):
+        timestamp = datetime.now()
+        timestamp = timestamp.strftime("[%Y-%m-%d %H:%M:%S]")
+        print(timestamp, end=' ')
+
         s_query = time.time()
         if self.use_context:
             print("Answering with your context:", question)