Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -50,6 +50,15 @@ def log_event(event):
50
  # =========================
51
  # Training Pipeline
52
  # =========================
 
 
 
 
 
 
 
 
 
53
  def train_model(model_name, dataset_name, epochs):
54
  try:
55
  log.info(f"Loading dataset: {dataset_name}")
 
50
  # =========================
51
  # Training Pipeline
52
  # =========================
53
+ import logging
54
+
55
+ logging.basicConfig(
56
+ filename=os.path.join("logs", "app.log"),
57
+ level=logging.INFO,
58
+ format="%(asctime)s [%(levelname)s] %(message)s"
59
+ )
60
+ log = logging.getLogger(__name__)
61
+
62
  def train_model(model_name, dataset_name, epochs):
63
  try:
64
  log.info(f"Loading dataset: {dataset_name}")