Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForConditionalGeneration, pipeline
+from transformers import BartTokenizer, BartForConditionalGeneration, pipeline
 import zipfile
 import os
 import nltk
@@ -28,9 +28,9 @@ print("Model Path:", model_path)
 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 using AutoTokenizer and AutoModelForConditionalGeneration
-    tokenizer = AutoTokenizer.from_pretrained(model_path)
-    model = AutoModelForConditionalGeneration.from_pretrained(model_path)
+    # Load the tokenizer and model from the extracted directory
+    tokenizer = BartTokenizer.from_pretrained(model_path)
+    model = BartForConditionalGeneration.from_pretrained(model_path)
 
     # Create a summarization pipeline
     summarizer = pipeline("summarization", model=model, tokenizer=tokenizer)