Lahiru Menikdiwela commited on
Commit
76446bf
·
1 Parent(s): 6238067

add logger for ip and op

Browse files
Files changed (1) hide show
  1. summarizer.py +2 -0
summarizer.py CHANGED
@@ -52,6 +52,7 @@ def summarizer_summarize(model_type,tokenizer, base_summarizer, text:str,summari
52
 
53
  #!!!!!!!!!!!!!!!!!!!!Changes to llama model
54
  input_text = text_to_summarize
 
55
  chat = [
56
  { "role": "user",
57
  "content": f"""
@@ -72,6 +73,7 @@ def summarizer_summarize(model_type,tokenizer, base_summarizer, text:str,summari
72
  summary = base_summary[0]
73
  # summary = pipe(text_to_summarize)[0]['generated_text']
74
  #!!!!!!!!!!!!!!!!!!!!!!!!!!!!!1Changes finished for llama model
 
75
 
76
 
77
  end = time.time()
 
52
 
53
  #!!!!!!!!!!!!!!!!!!!!Changes to llama model
54
  input_text = text_to_summarize
55
+ logger.info("Input", input_text)
56
  chat = [
57
  { "role": "user",
58
  "content": f"""
 
73
  summary = base_summary[0]
74
  # summary = pipe(text_to_summarize)[0]['generated_text']
75
  #!!!!!!!!!!!!!!!!!!!!!!!!!!!!!1Changes finished for llama model
76
+ logger.info("Output: "+summary)
77
 
78
 
79
  end = time.time()