Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import T5Tokenizer, AutoModelForSeq2SeqLM, pipeline
+from transformers import BartTokenizer, BartForConditionalGeneration, pipeline
 import zipfile
 import os
 import nltk
@@ -26,8 +26,8 @@ if not os.path.exists(model_path):
     st.error(f"Model directory {model_path} does not exist or is incorrect.")
 else:
     # Load the tokenizer and model
-    tokenizer = T5Tokenizer.from_pretrained(model_path)
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
+    tokenizer = BartTokenizer.from_pretrained(model_name)
+    model = BartForConditionalGeneration.from_pretrained(model_name)
 
     # Create a summarization pipeline
     summarizer = pipeline("summarization", model=model, tokenizer=tokenizer)