AMabInsta commited on
Commit
757d63f
·
1 Parent(s): dd42569

fix: setup logging correctly for logger

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -49,11 +49,14 @@ gr.set_static_paths(paths=[Path.cwd().absolute()/"assets"])
49
  log_file = "/tmp/instanovo_gradio_log.txt"
50
  Path(log_file).touch()
51
 
 
 
52
  file_handler = logging.FileHandler(log_file)
53
- file_handler.setLevel(logging.DEBUG)
 
 
 
54
 
55
- logger = logging.getLogger("instanovo")
56
- logger.setLevel(logging.DEBUG)
57
 
58
  def load_model_and_knapsack():
59
  """Loads the InstaNovo model and generates/loads the knapsack."""
 
49
  log_file = "/tmp/instanovo_gradio_log.txt"
50
  Path(log_file).touch()
51
 
52
+ logger = logging.getLogger("instanovo")
53
+ logger.setLevel(logging.INFO)
54
  file_handler = logging.FileHandler(log_file)
55
+ file_handler.setLevel(logging.INFO)
56
+ formatter = logging.Formatter("%(asctime)s - %(levelname)s - %(message)s")
57
+ file_handler.setFormatter(formatter)
58
+ logger.addHandler(file_handler)
59
 
 
 
60
 
61
  def load_model_and_knapsack():
62
  """Loads the InstaNovo model and generates/loads the knapsack."""