GurgenGulay commited on
Commit
656c17d
·
verified ·
1 Parent(s): e854524

Update text_analysis.py

Browse files
Files changed (1) hide show
  1. text_analysis.py +5 -9
text_analysis.py CHANGED
@@ -1,16 +1,13 @@
1
  from transformers import pipeline
2
 
3
- # Modeli bir kez yükleyip her seferinde yeniden yüklememek için global değişken kullanabiliriz.
4
  pipe = pipeline("text2text-generation", model="google-t5/t5-base", device="cpu")
5
- pipe.model.config.pad_token_id = pipe.tokenizer.eos_token_id # pad_token_id ayarı
6
 
7
  def generate_lesson_from_transcript(doc_text):
8
- """
9
- Metin girişinden ders anlatımı üretir ve çıktı olarak metin ile dosya döndürür.
10
- """
11
  try:
12
  generated_text = pipe(doc_text, max_length=100, truncation=True)[0]['generated_text']
13
- output_path = "/tmp/generated_output.txt" # Web ortamında /tmp gibi geçici bir dizin kullanabilirsiniz
14
 
15
  with open(output_path, "w") as file:
16
  file.write(generated_text)
@@ -18,6 +15,5 @@ def generate_lesson_from_transcript(doc_text):
18
  return generated_text, output_path
19
 
20
  except Exception as e:
21
- # Hata mesajını daha ayrıntılı yazdırabiliriz.
22
- print(f"Bir hata oluştu: {str(e)}")
23
- return "Bir hata oluştu", None
 
1
  from transformers import pipeline
2
 
3
+ # Initialize the pipeline globally to avoid reloading the model each time
4
  pipe = pipeline("text2text-generation", model="google-t5/t5-base", device="cpu")
5
+ pipe.model.config.pad_token_id = pipe.tokenizer.eos_token_id # Set pad_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" # Use a temporary directory for web environments
11
 
12
  with open(output_path, "w") as file:
13
  file.write(generated_text)
 
15
  return generated_text, output_path
16
 
17
  except Exception as e:
18
+ print(f"An error occurred: {str(e)}")
19
+ return "An error occurred", None