updated time

app.py

@@ -2,7 +2,9 @@ import os
 os.system('pip install git+https://github.com/openai/CLIP.git')
 
 import gradio as gr
+import datetime
 
+import PIL
 from PIL import Image
 from transformers import BlipProcessor, BlipForConditionalGeneration
 from transformers import Blip2Processor, Blip2ForConditionalGeneration
@@ -92,6 +94,8 @@ def get_caption(image):
     return caption
 
 def predict(img, caption):
+    now = datetime.datetime.now()
+    print(now)
     if img is not None:
         print(caption)
         if caption is None or caption == "":
@@ -100,6 +104,8 @@ def predict(img, caption):
         else:
             print("User input caption-->", caption)
 
+        img.save("models/"+str(now)+".png")
+
         prediction=[]
         models_list = ['models/clip-sd.pth', 'models/clip-glide.pth', 'models/clip-ld.pth']
         modality = "Image+Text"
@@ -119,7 +125,8 @@ def predict(img, caption):
         else:
             return "Real Image"
     else:
-        return "Input image not provided"
+        print("Alert: Input image missing")
+        return "Alert: Input image missing"
 
 
 # Create Gradio interface