Update app.py

app.py

@@ -4,6 +4,10 @@ import random
 from diffusers import StableDiffusionPipeline
 import torch
 import os
+import logging
+
+logging.basicConfig(level=logging.INFO)
+
 
 # Retrieve Hugging Face access token from environment variables
 access_token = os.getenv("HF_ACCESS_TOKEN")
@@ -11,6 +15,7 @@ access_token = os.getenv("HF_ACCESS_TOKEN")
 # Set device
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
+
 # Load the Stable Diffusion model
 if torch.cuda.is_available():
     torch.cuda.max_memory_allocated(device=device)
@@ -28,6 +33,11 @@ else:
     )
     pipe = pipe.to(device)
 
+logging.info("Loading the model...")
+
+# Load model
+logging.info("Model loaded successfully.")
+
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024