Lahiru Menikdiwela
commited on
Commit
·
76446bf
1
Parent(s):
6238067
add logger for ip and op
Browse files- 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()
|