Update app.py
Browse files
app.py
CHANGED
@@ -60,11 +60,13 @@ def speech_answer(audio_file_path):
|
|
60 |
# ------------------------------
|
61 |
# Summarizer
|
62 |
# ------------------------------
|
|
|
|
|
|
|
|
|
63 |
def summarize_text(text):
|
64 |
-
|
65 |
-
|
66 |
-
summary_ids = model_qg.generate(input_ids, max_length=150, min_length=30, length_penalty=5.0, num_beams=2)
|
67 |
-
return tokenizer_qg.decode(summary_ids[0], skip_special_tokens=True)
|
68 |
|
69 |
# ------------------------------
|
70 |
# Engagement Predictor (Mock)
|
|
|
60 |
# ------------------------------
|
61 |
# Summarizer
|
62 |
# ------------------------------
|
63 |
+
from transformers import pipeline
|
64 |
+
|
65 |
+
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
|
66 |
+
|
67 |
def summarize_text(text):
|
68 |
+
summary = summarizer(text, max_length=100, min_length=30, do_sample=False)
|
69 |
+
return summary[0]['summary_text']
|
|
|
|
|
70 |
|
71 |
# ------------------------------
|
72 |
# Engagement Predictor (Mock)
|