Update app.py

app.py

@@ -7,11 +7,11 @@ import numpy as np
 from nsfw_detector import NSFWDetector, create_error_image
 from PIL import Image
 import time
-import logging
+# import logging
 from threading import Timer
 
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
+# logging.basicConfig(level=logging.INFO)
+# logger = logging.getLogger(__name__)
 
 # Global variables
 global_model = None
@@ -81,7 +81,7 @@ def unload_model():
     global global_model, last_use_time
     current_time = time.time()
     if last_use_time and (current_time - last_use_time) >= TIMEOUT_SECONDS:
-        logger.info("Unloading model due to inactivity...")
+        # logger.info("Unloading model due to inactivity...")
         global_model = None
         torch.cuda.empty_cache()
         return "Model unloaded due to inactivity"
@@ -104,7 +104,7 @@ def generate_image(prompt, height, width, guidance_scale, pag_guidance_scale, nu
             
         # Load model if needed
         if global_model is None:
-            logger.info("Loading model...")
+            # logger.info("Loading model...")
             global_model = SanaPipeline("configs/sana_config/1024ms/Sana_1600M_img1024.yaml")
             global_model.from_pretrained("hf://Efficient-Large-Model/Sana_1600M_1024px/checkpoints/Sana_1600M_1024px.pth")
             
@@ -134,11 +134,11 @@ def generate_image(prompt, height, width, guidance_scale, pag_guidance_scale, nu
         if category == "SAFE":
             return image
         else:
-            logger.warning(f"NSFW content detected ({category} with {confidence:.2f}% confidence)")
+            # logger.warning(f"NSFW content detected ({category} with {confidence:.2f}% confidence)")
             return create_error_image()
         
     except Exception as e:
-        logger.error(f"Error in generate_image: {str(e)}")
+        # logger.error(f"Error in generate_image: {str(e)}")
         raise gr.Error(f"Generation failed: {str(e)}")
 
 # Gradio Interface