aditi2222 commited on
Commit
c858aa8
·
1 Parent(s): 9774338

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -3
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(translated_text):
10
- input_ = str(text) + ' </s>'
11
  # generate summary
12
- input_ids = tokenizer.encode(translated_text, return_tensors='pt')
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)