Update app.py

app.py

@@ -6,6 +6,8 @@ import gradio as gr
 from transformers import CLIPProcessor, CLIPModel
 import numpy as np
 import time
+import os
+from datetime import datetime
 
 # Replace with your own API key
 STABLE_DIFFUSION_API_KEY = "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"
@@ -74,6 +76,9 @@ def generate_art(mood):
         break
 
     image = Image.open(BytesIO(response.content))
+    timestamp = datetime.now().strftime("%Y%m%d-%H%M%S")
+    image.save(f"generated_images/{timestamp}_{mood}.jpg")
+    
 
     return image