Spaces:

Abhishek-D7
/

MULTILINGUAL-TEXT-SUMMARIZER-AND-TRANSLATOR

App Files Files Community

Abhishek-D7 commited on Jul 29

Commit

fcf5e2e

•

1 Parent(s): abc5e07

Update app.py

Files changed (1) hide show

app.py +3 -3

app.py CHANGED Viewed

@@ -1,10 +1,10 @@
 import streamlit as st
-from transformers import AutoTokenizer, BartForConditionalGeneration, MarianMTModel, MarianTokenizer
-# Load summarization and translation models
 def load_summarization_model():
     model = BartForConditionalGeneration.from_pretrained('facebook/bart-large-cnn')
-    tokenizer = AutoTokenizer.from_pretrained('facebook/bart-large-cnn')
     return model, tokenizer
 def load_translation_models():

 import streamlit as st
+from transformers import BartTokenizer, BartForConditionalGeneration, MarianMTModel, MarianTokenizer
 def load_summarization_model():
     model = BartForConditionalGeneration.from_pretrained('facebook/bart-large-cnn')
+    tokenizer = BartTokenizer.from_pretrained('facebook/bart-large-cnn')
     return model, tokenizer
 def load_translation_models():