varl42 commited on
Commit
5025ff4
·
1 Parent(s): 865e98b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=="Summary",audio_summary=="Audio"],
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",