Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,13 +3,29 @@ from transformers import BartTokenizer, BartForConditionalGeneration
|
|
3 |
import gradio as gr
|
4 |
from transformers import AutoTokenizer, AutoModelWithLMHead, TranslationPipeline
|
5 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
model = BartForConditionalGeneration.from_pretrained("facebook/bart-large-cnn")
|
7 |
tokenizer = BartTokenizer.from_pretrained("facebook/bart-large-cnn")
|
8 |
|
9 |
-
def article_summarization(
|
10 |
-
input_ = str(
|
11 |
# generate summary
|
12 |
-
input_ids = tokenizer.encode(
|
13 |
summary_ids = model.generate(input_ids,
|
14 |
min_length=20,
|
15 |
max_length=12000)
|
|
|
3 |
import gradio as gr
|
4 |
from transformers import AutoTokenizer, AutoModelWithLMHead, TranslationPipeline
|
5 |
|
6 |
+
|
7 |
+
|
8 |
+
from transformers import pipeline
|
9 |
+
import gradio as gr
|
10 |
+
pipe= pipeline('text2text-generation', model="facebook/m2m100_418M")
|
11 |
+
def generate_text(inp):
|
12 |
+
output=pipe(inp, forced_bos_token_id=pipe.tokenizer.get_lang_id('en'))
|
13 |
+
tln=output[0]
|
14 |
+
for item in tln:
|
15 |
+
result=tln[item]
|
16 |
+
return result
|
17 |
+
#Gradio Interface
|
18 |
+
|
19 |
+
|
20 |
+
|
21 |
+
|
22 |
model = BartForConditionalGeneration.from_pretrained("facebook/bart-large-cnn")
|
23 |
tokenizer = BartTokenizer.from_pretrained("facebook/bart-large-cnn")
|
24 |
|
25 |
+
def article_summarization(result):
|
26 |
+
input_ = str(result) + ' </s>'
|
27 |
# generate summary
|
28 |
+
input_ids = tokenizer.encode(result, return_tensors='pt')
|
29 |
summary_ids = model.generate(input_ids,
|
30 |
min_length=20,
|
31 |
max_length=12000)
|