GurgenGulay commited on
Commit
c2386f3
·
verified ·
1 Parent(s): 1deb054

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +10 -4
utils.py CHANGED
@@ -1,22 +1,28 @@
 
1
  from transformers import pipeline
2
 
 
 
 
3
 
4
  pipe = pipeline("text2text-generation", model="google-t5/t5-base", device="cpu")
5
  pipe.model.config.pad_token_id = pipe.tokenizer.eos_token_id
6
 
7
  def generate_lesson_from_transcript(doc_text):
8
  try:
 
9
  generated_text = pipe(doc_text, max_length=100, truncation=True)[0]['generated_text']
10
- output_path = "/tmp/generated_output.txt"
11
 
12
  with open(output_path, "w") as file:
13
  file.write(generated_text)
14
 
 
15
  return generated_text, output_path
16
-
17
  except Exception as e:
18
- print(f"Bir hata oluştu: {str(e)}")
19
- return "Bir hata oluştu", None
20
 
21
  def split_text_into_chunks(text, chunk_size=1000):
22
  words = text.split()
 
1
+ import logging
2
  from transformers import pipeline
3
 
4
+ # Logging Ayarları
5
+ logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
6
+ logger = logging.getLogger(__name__)
7
 
8
  pipe = pipeline("text2text-generation", model="google-t5/t5-base", device="cpu")
9
  pipe.model.config.pad_token_id = pipe.tokenizer.eos_token_id
10
 
11
  def generate_lesson_from_transcript(doc_text):
12
  try:
13
+ logger.info("Generating lesson from transcript.")
14
  generated_text = pipe(doc_text, max_length=100, truncation=True)[0]['generated_text']
15
+ output_path = "/tmp/generated_output.txt"
16
 
17
  with open(output_path, "w") as file:
18
  file.write(generated_text)
19
 
20
+ logger.info(f"Lesson generation successful. Output saved at: {output_path}")
21
  return generated_text, output_path
22
+
23
  except Exception as e:
24
+ logger.error(f"Error occurred during lesson generation: {str(e)}")
25
+ return "An error occurred", None
26
 
27
  def split_text_into_chunks(text, chunk_size=1000):
28
  words = text.split()