GurgenGulay commited on
Commit
b11e771
·
verified ·
1 Parent(s): 000e05e

Update text_analysis.py

Browse files
Files changed (1) hide show
  1. text_analysis.py +3 -4
text_analysis.py CHANGED
@@ -1,8 +1,8 @@
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
  """
@@ -10,7 +10,7 @@ def generate_lesson_from_transcript(doc_text):
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 +18,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
+
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
  """
 
10
  """
11
  try:
12
  generated_text = pipe(doc_text, max_length=100, truncation=True)[0]['generated_text']
13
+ output_path = "/tmp/generated_output.txt"
14
 
15
  with open(output_path, "w") as file:
16
  file.write(generated_text)
 
18
  return generated_text, output_path
19
 
20
  except Exception as e:
 
21
  print(f"Bir hata oluştu: {str(e)}")
22
  return "Bir hata oluştu", None