KAPtechies commited on
Commit
a8051c2
·
1 Parent(s): b1bfe7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -9,9 +9,11 @@ news=" ".join(result)
9
  from transformers import AutoTokenizer
10
  tokenizer=AutoTokenizer.from_pretrained("facebook/mbart-large-50-one-to-many-mmt",use_fast=False)
11
  from transformers import MBartForConditionalGeneration
 
12
  # download and save model
13
  model=MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-one-to-many-mmt")
14
  input_text=[news]
 
15
  # convert sentences to tensors
16
  model_inputs=tokenizer(input_text,return_tensors="pt",padding=True,truncation=True)
17
  # translate from English to Hindi
 
9
  from transformers import AutoTokenizer
10
  tokenizer=AutoTokenizer.from_pretrained("facebook/mbart-large-50-one-to-many-mmt",use_fast=False)
11
  from transformers import MBartForConditionalGeneration
12
+
13
  # download and save model
14
  model=MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-one-to-many-mmt")
15
  input_text=[news]
16
+
17
  # convert sentences to tensors
18
  model_inputs=tokenizer(input_text,return_tensors="pt",padding=True,truncation=True)
19
  # translate from English to Hindi