Update core/image_generator.py

core/image_generator.py

@@ -5,6 +5,9 @@ from diffusers import StableDiffusionXLPipeline
 from huggingface_hub import hf_hub_download
 from pathlib import Path
 from typing import List
+from io import BytesIO
+import base64
+from PIL import Image
 
 
 # ---------------- HF CACHE FIX ----------------