rahul7star commited on
Commit
a20dc48
·
verified ·
1 Parent(s): 0aa6485

Update simple_app.py

Browse files
Files changed (1) hide show
  1. simple_app.py +1 -1
simple_app.py CHANGED
@@ -102,7 +102,7 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
102
  if "INFO:" in stripped_line:
103
  parts = stripped_line.split("INFO:", 1)
104
  msg = parts[1].strip() if len(parts) > 1 else ""
105
- print(stripped_line) # Log the message
106
 
107
  # For the first 4 INFO messages, simply count them.
108
  if processed_steps < irrelevant_steps:
 
102
  if "INFO:" in stripped_line:
103
  parts = stripped_line.split("INFO:", 1)
104
  msg = parts[1].strip() if len(parts) > 1 else ""
105
+ print(f"[INFO]: {msg}") # Log the message
106
 
107
  # For the first 4 INFO messages, simply count them.
108
  if processed_steps < irrelevant_steps: