Update app.py

app.py

@@ -38,15 +38,15 @@ def bart_summarize(text):
     return pp
 
 #Encoder-Decoder
-# def encoder_decoder(text):
-#   model = EncoderDecoderModel.from_pretrained("patrickvonplaten/bert2bert_cnn_daily_mail")
-#   tokenizer = AutoTokenizer.from_pretrained("patrickvonplaten/bert2bert_cnn_daily_mail")
-#   # let's perform inference on a long piece of text
-#   input_ids = tokenizer(text, return_tensors="pt").input_ids
-#   # autoregressively generate summary (uses greedy decoding by default) 
-#   generated_ids = model.generate(input_ids)
-#   generated_text = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
-#   return generated_text
+def encoder_decoder(text):
+    model = EncoderDecoderModel.from_pretrained("patrickvonplaten/bert2bert_cnn_daily_mail")
+    tokenizer = AutoTokenizer.from_pretrained("patrickvonplaten/bert2bert_cnn_daily_mail")
+    # let's perform inference on a long piece of text
+    input_ids = tokenizer(text, return_tensors="pt").input_ids
+    # autoregressively generate summary (uses greedy decoding by default) 
+    generated_ids = model.generate(input_ids)
+    generated_text = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
+    return generated_text
 
 
 #Input text
@@ -77,8 +77,9 @@ if button:
         elif model == "BART":
             st.write("You have selected BART model.")
             summary = bart_summarize(text)
-        #elif model == "Encoder-Decoder":
-            #st.write("You have selected Encoder-Decoder model.")
+        elif model == "Encoder-Decoder":
+            st.write("You have selected Encoder-Decoder model.")
+            summary = encoder_decoder(text)
 
         #st.toast("Please wait while we summarize your text.")
         #with st.spinner("Summarizing..."):