added timestamp to logs

gradio_app.py

@@ -1,5 +1,7 @@
 import os
+import time
 import csv
+import datetime
 import gradio
 from gradio import utils
 import huggingface_hub
@@ -15,7 +17,7 @@ hf_token = os.getenv("HF_TOKEN")
 
 dataset_dir = "logs"
 
-headers = ["input", "output"]
+headers = ["input", "output", "timestamp", "elapsed"]
 
 
 repo = huggingface_hub.Repository(
@@ -24,7 +26,7 @@ repo = huggingface_hub.Repository(
     token=hf_token,
 )
 
-def log_record(input, output):
+def log_record(vals):
     repo.git_pull(lfs=True)
     
     log_file = Path(dataset_dir) / "data.csv"
@@ -37,7 +39,7 @@ def log_record(input, output):
         if is_new:
             writer.writerow(utils.sanitize_list_for_csv(headers))
 
-        writer.writerow(utils.sanitize_list_for_csv([input, output]))
+        writer.writerow(utils.sanitize_list_for_csv(vals))
     
     with open(log_file, "r", encoding="utf-8") as csvfile:
         line_count = len([None for _ in csv.reader(csvfile)]) - 1
@@ -49,13 +51,19 @@ def predict(sentence):
     
     print(sentence)
     
+    timestamp = datetime.datetime.now().isoformat()
+    
+    start_time = time.time()
+    
     predictions = model.evaluate([sentence])
     
+    elapsed_time = time.time() - start_time
+    
     print(f"Predictions: {predictions}")
     
     output = classes[predictions[0]]
     
-    log_record(sentence, output)
+    log_record([sentence, output, timestamp, str(elapsed_time)])
     
     return output