Update app.py

app.py

@@ -4,6 +4,7 @@ random.seed(999)
 import torch
 from torchvision.transforms import transforms
 import gradio as gr
+from datetime import datetime
 
 model = torch.load('model.pth', map_location=torch.device('cpu'))
 model.eval()
@@ -49,6 +50,8 @@ def create_tags(image, threshold):
     # print("Before adding implicated tags, there are " + str(len(temp)) + " tags")
     temp = [t[0] for t in temp]
     text_no_impl = " ".join(temp)
+    current_datetime = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+    print(f"{current_datetime}: finished.")
     return text_no_impl, tag_score
 
 demo = gr.Interface(