Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ def summarize_text(text):
|
|
31 |
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
|
32 |
|
33 |
# Tokenize abstract
|
34 |
-
inputs = tokenizer(abstract, return_tensors="pt", truncation=True)
|
35 |
|
36 |
# Generate summary
|
37 |
summary_ids = model.generate(inputs['input_ids'], num_beams=2, max_length=42, min_length=22, do_sample=True, early_stopping=True)
|
@@ -60,7 +60,7 @@ audio_summary = gr.Audio()
|
|
60 |
iface = gr.Interface(
|
61 |
fn=audio_pdf,
|
62 |
inputs=inputs,
|
63 |
-
outputs=[summary_text
|
64 |
title="PDF Audio Summarizer 📻",
|
65 |
description="App that converts an abstract into audio",
|
66 |
examples=["Attention_is_all_you_need.pdf",
|
|
|
31 |
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
|
32 |
|
33 |
# Tokenize abstract
|
34 |
+
inputs = tokenizer(abstract, max_length=1024, return_tensors="pt", truncation=True)
|
35 |
|
36 |
# Generate summary
|
37 |
summary_ids = model.generate(inputs['input_ids'], num_beams=2, max_length=42, min_length=22, do_sample=True, early_stopping=True)
|
|
|
60 |
iface = gr.Interface(
|
61 |
fn=audio_pdf,
|
62 |
inputs=inputs,
|
63 |
+
outputs=[summary_text,audio_summary],
|
64 |
title="PDF Audio Summarizer 📻",
|
65 |
description="App that converts an abstract into audio",
|
66 |
examples=["Attention_is_all_you_need.pdf",
|