Amelia-James commited on
Commit
17d8c48
·
verified ·
1 Parent(s): 06afc91

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -0
app.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import BartForConditionalGeneration, BartTokenizer
2
+
3
+ model = BartForConditionalGeneration.from_pretrained('facebook/bart-large-cnn')
4
+ tokenizer = BartTokenizer.from_pretrained('facebook/bart-large-cnn')
5
+
6
+ def summarize_text(text):
7
+ inputs = tokenizer([text], max_length=1024, return_tensors='pt', truncation=True)
8
+ summary_ids = model.generate(inputs['input_ids'], num_beams=4, max_length=200, early_stopping=True)
9
+ summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
10
+ return summary
11
+
12
+ import gradio as gr
13
+
14
+ interface = gr.Interface(fn=summarize_text,
15
+ inputs="text",
16
+ outputs="text",
17
+ title="Text Summarization Tool",
18
+ description="Enter the text you want to summarize.")
19
+ interface.launch()