thomasanto7001 commited on
Commit
0fe9459
·
verified ·
1 Parent(s): d1ad8ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -7
app.py CHANGED
@@ -43,14 +43,27 @@ def generate_summary(text, max_len=130):
43
  summary += summarizer(chunk, max_length=max_len, min_length=30, do_sample=False)[0]["summary_text"] + " "
44
  return summary.strip()
45
 
46
- def generate_subtitles(text):
 
47
  sentences = sent_tokenize(text)
48
- subtitles = []
49
- for i, sentence in enumerate(sentences):
50
- start_time = i * 5
51
- end_time = start_time + 5
52
- subtitles.append(f"{i+1}\n00:00:{start_time:02},000 --> 00:00:{end_time:02},000\n{sentence}\n")
53
- return "\n".join(subtitles)
 
 
 
 
 
 
 
 
 
 
 
 
54
 
55
  def generate_quiz(text, num_questions=5):
56
  sentences = sent_tokenize(text)
 
43
  summary += summarizer(chunk, max_length=max_len, min_length=30, do_sample=False)[0]["summary_text"] + " "
44
  return summary.strip()
45
 
46
+ def generate_summary(text, max_len=130, min_len=30):
47
+ summarizer = pipeline("summarization")
48
  sentences = sent_tokenize(text)
49
+ chunks = [' '.join(sentences[i:i + 10]) for i in range(0, len(sentences), 10)]
50
+ summary = ""
51
+
52
+ for chunk in chunks:
53
+ input_len = len(chunk.split())
54
+ dynamic_max_len = min(max_len, max(20, input_len // 2))
55
+ dynamic_min_len = min(min_len, dynamic_max_len - 5) # Ensure min_len < max_len
56
+
57
+ result = summarizer(
58
+ chunk,
59
+ max_length=dynamic_max_len,
60
+ min_length=dynamic_min_len,
61
+ do_sample=False
62
+ )[0]["summary_text"]
63
+
64
+ summary += result + " "
65
+
66
+ return summary.strip()
67
 
68
  def generate_quiz(text, num_questions=5):
69
  sentences = sent_tokenize(text)