Update app.py

add log of the file

app.py

@@ -4,6 +4,7 @@ import torch
 from PIL import Image
 from diffusers import DiffusionPipeline
 import random
+from datetime import datetime
 
 # Initialize the base model and specific LoRA
 base_model = "black-forest-labs/FLUX.1-dev"
@@ -47,6 +48,19 @@ def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora
         joint_attention_kwargs={"scale": lora_scale},
     ).images[0]
 
+    # Save the image to a file with a unique name in /tmp directory
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    image_filename = f"generated_image_{timestamp}.png"
+    image_path = os.path.join("/tmp/gradio", image_filename)
+    image.save(image_path)
+
+    # Construct the URL to access the image
+    space_url = "https://killwithabass-flux-1-dev-lora-androflux.hf.space"  # Replace with your actual space URL
+    image_url = f"{space_url}/file={image_path}"
+
+    # Log the file URL
+    print(f"Image URL: {image_url}")
+
     # Final update (100%)
     progress(100, "Completed!")