Update app.py

app.py

@@ -22,6 +22,7 @@ logger = logging.getLogger(__name__)
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 logger.info(f"Using device: {device}")
 # Set up custom logging
+# Set up custom logging
 class CustomFormatter(logging.Formatter):
     """Custom formatter with colors and better formatting"""
     grey = "\x1b[38;21m"
@@ -31,30 +32,34 @@ class CustomFormatter(logging.Formatter):
     bold_red = "\x1b[31;1m"
     reset = "\x1b[0m"
 
-    format_str = "%(message)s"
-
-    FORMATS = {
-        logging.DEBUG: blue + format_str + reset,
-        logging.INFO: grey + format_str + reset,
-        logging.WARNING: yellow + format_str + reset,
-        logging.ERROR: red + format_str + reset,
-        logging.CRITICAL: bold_red + format_str + reset
-    }
-
     def format(self, record):
-        log_fmt = self.FORMATS.get(record.levelno)
-        formatter = logging.Formatter(log_fmt)
-        return formatter.format(record)
+        # Remove the logger name from the output
+        if record.levelno == logging.DEBUG:
+            return f"{self.blue}{record.getMessage()}{self.reset}"
+        elif record.levelno == logging.INFO:
+            return f"{self.grey}{record.getMessage()}{self.reset}"
+        elif record.levelno == logging.WARNING:
+            return f"{self.yellow}{record.getMessage()}{self.reset}"
+        elif record.levelno == logging.ERROR:
+            return f"{self.red}{record.getMessage()}{self.reset}"
+        elif record.levelno == logging.CRITICAL:
+            return f"{self.bold_red}{record.getMessage()}{self.reset}"
+        return record.getMessage()
 
 # Setup logger
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.DEBUG)
 
+# Remove any existing handlers
+logger.handlers = []
+
 # Create console handler with custom formatter
 ch = logging.StreamHandler()
 ch.setLevel(logging.DEBUG)
 ch.setFormatter(CustomFormatter())
-logger.handlers = [ch]
+logger.addHandler(ch)
+
+
 
 # Model initialization
 model_name = "SamanthaStorm/tether-multilabel-v4"