update

app.py

@@ -1,15 +1,13 @@
 import os
 import sys
-import logging
 sys.path.append('app/')
-logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
-logger = logging.getLogger()
 
 import torch
 import spaces
 import safetensors
 import gradio as gr
 from PIL import Image
+from loguru import logger
 from torchvision import transforms
 from huggingface_hub import hf_hub_download, login
 from diffusers import FluxPipeline, FluxTransformer2DModel

requirements.txt

@@ -4,4 +4,5 @@ invisible_watermark
 torch
 transformers
 torchvision
-timm
+timm
+loguru